diff --git a/rl_coach/graph_managers/graph_manager.py b/rl_coach/graph_managers/graph_manager.py
index aa84dd3..53525ea 100644
--- a/rl_coach/graph_managers/graph_manager.py
+++ b/rl_coach/graph_managers/graph_manager.py
@@ -220,7 +220,7 @@ class GraphManager(object):
             self.set_session(self.sess)
         else:
             self.variables_to_restore = tf.global_variables()
-            self.variables_to_restore = [v for v in self.variables_to_restore if '/online' in v.name]
+            # self.variables_to_restore = [v for v in self.variables_to_restore if '/online' in v.name] TODO: is this necessary?
             self.checkpoint_saver = tf.train.Saver(self.variables_to_restore)
 
             # regular session