diff --git a/video_prediction_tools/main_scripts/main_train_models.py b/video_prediction_tools/main_scripts/main_train_models.py
index e092a8558fbebaf6bea68aff9dc63c692a3a2c6b..300291ba4b3771335d9eb157379ee8d6be75980e 100644
--- a/video_prediction_tools/main_scripts/main_train_models.py
+++ b/video_prediction_tools/main_scripts/main_train_models.py
@@ -405,7 +405,7 @@ class TrainModel(object):
 
         return self.val_fetches
 
-    def generate_fetches(self, fetch_list, nest_element=None):
+    def generate_fetches(self, fetch_list):
         """
         Generates dictionary of fetches from video model instance
         :param fetch_list: list of attributes of video model instance that are of particular interest;