diff --git a/train.py b/train.py
index 0f457fa..a2acfe1 100644
--- a/train.py
+++ b/train.py
@@ -103,8 +103,6 @@ def train(agent_name="Blue", gamma=0.99, epsilon=1, lr=0.0001,
             env = ChallengeWrapper(agent_name="Blue",
                                    env=CybORG(PATH, 'sim', agents=agent_list[agent_counter % len(agent_list)]))
     except KeyboardInterrupt:
-        plot_learning_curve(steps_array, scores, eps_history, FIGURE_FILE)
-
         agent.save_models()