diff --git a/neural_network/back_propagation_neural_network.py b/neural_network/back_propagation_neural_network.py
index 9dd112115f5e..2001a3aac9dd 100644
--- a/neural_network/back_propagation_neural_network.py
+++ b/neural_network/back_propagation_neural_network.py
@@ -163,7 +163,7 @@ def cal_loss(self, ydata, ydata_):
 
     def plot_loss(self):
         if self.ax_loss.lines:
-            self.ax_loss.lines.remove(self.ax_loss.lines[0])
+            self.ax_loss.lines[0].remove()
         self.ax_loss.plot(self.train_mse, "r-")
         plt.ion()
         plt.xlabel("step")