diff --git a/rl_coach/graph_managers/graph_manager.py b/rl_coach/graph_managers/graph_manager.py
index 6fc8fc1..56911d2 100644
--- a/rl_coach/graph_managers/graph_manager.py
+++ b/rl_coach/graph_managers/graph_manager.py
@@ -288,8 +288,7 @@ class GraphManager(object):
         count_end = self.total_steps_counters[RunPhase.TRAIN][TrainingSteps] + steps.num_steps
         while self.total_steps_counters[RunPhase.TRAIN][TrainingSteps] < count_end:
             self.total_steps_counters[RunPhase.TRAIN][TrainingSteps] += 1
-            losses = [manager.train() for manager in self.level_managers]
-            # self.loss.add_sample(loss)
+            [manager.train() for manager in self.level_managers]
 
     def reset_internal_state(self, force_environment_reset=False) -> None:
         """