diff --git a/video_prediction_tools/model_modules/video_prediction/models/convLSTM_GAN_model.py b/video_prediction_tools/model_modules/video_prediction/models/convLSTM_GAN_model.py
index 972f95576a90563241f6eede2c4d973115092d06..092cf81a0db1adeb3c3bac65cb3bc92e54e3e4c5 100644
--- a/video_prediction_tools/model_modules/video_prediction/models/convLSTM_GAN_model.py
+++ b/video_prediction_tools/model_modules/video_prediction/models/convLSTM_GAN_model.py
@@ -2,7 +2,7 @@ __email__ = "b.gong@fz-juelich.de"
 __author__ = "Bing Gong,Yanji"
 __date__ = "2021-04-13"
 
-from model_helpers import set_and_check_pred_frames
+from model_modules.video_prediction.models.model_helpers import set_and_check_pred_frames
 import tensorflow as tf
 from model_modules.video_prediction.layers import layer_def as ld
 from model_modules.video_prediction.layers.BasicConvLSTMCell import BasicConvLSTMCell
diff --git a/video_prediction_tools/model_modules/video_prediction/models/mcnet_model.py b/video_prediction_tools/model_modules/video_prediction/models/mcnet_model.py
index 61fdb9121c3f127a0fb873df546c629fb679ff79..2c755417c27bca22b2a20dc6816add43f160c5a6 100644
--- a/video_prediction_tools/model_modules/video_prediction/models/mcnet_model.py
+++ b/video_prediction_tools/model_modules/video_prediction/models/mcnet_model.py
@@ -7,7 +7,7 @@ import itertools
 import numpy as np
 import tensorflow as tf
 
-from model_helpers import set_and_check_pred_frames
+from model_modules.video_prediction.models.model_helpers import set_and_check_pred_frames
 from model_modules.video_prediction.models import BaseVideoPredictionModel
 from model_modules.video_prediction.ops import dense, pad2d, conv2d, flatten, tile_concat
 from model_modules.video_prediction.layers.BasicConvLSTMCell import BasicConvLSTMCell
diff --git a/video_prediction_tools/model_modules/video_prediction/models/vanilla_GAN_model.py b/video_prediction_tools/model_modules/video_prediction/models/vanilla_GAN_model.py
index 747e19334e37d001439529f7a449a69b1a3a56f8..112eaf31e1d9961e402ddbac1a91b37a6b7b9b90 100644
--- a/video_prediction_tools/model_modules/video_prediction/models/vanilla_GAN_model.py
+++ b/video_prediction_tools/model_modules/video_prediction/models/vanilla_GAN_model.py
@@ -13,7 +13,7 @@ This code implement take the following as references:
 
 import tensorflow as tf
 
-from model_helpers import set_and_check_pred_frames
+from model_modules.video_prediction.models.model_helpers import set_and_check_pred_frames
 from model_modules.video_prediction.layers import layer_def as ld
 from tensorflow.contrib.training import HParams
 
diff --git a/video_prediction_tools/model_modules/video_prediction/models/vanilla_convLSTM_model.py b/video_prediction_tools/model_modules/video_prediction/models/vanilla_convLSTM_model.py
index 1780b2e8439341320fe5726dab8f7174225a5956..570ece3e8e752a0520bf1593bc5e0b922cece3a0 100644
--- a/video_prediction_tools/model_modules/video_prediction/models/vanilla_convLSTM_model.py
+++ b/video_prediction_tools/model_modules/video_prediction/models/vanilla_convLSTM_model.py
@@ -2,7 +2,7 @@ __email__ = "b.gong@fz-juelich.de"
 __author__ = "Bing Gong, Scarlet Stadtler,Michael Langguth"
 __date__ = "2020-11-05"
 
-from model_helpers import set_and_check_pred_frames
+from model_modules.video_prediction.models.model_helpers import set_and_check_pred_frames
 import tensorflow as tf
 from model_modules.video_prediction.layers import layer_def as ld
 from model_modules.video_prediction.layers.BasicConvLSTMCell import BasicConvLSTMCell
diff --git a/video_prediction_tools/model_modules/video_prediction/models/vanilla_vae_model.py b/video_prediction_tools/model_modules/video_prediction/models/vanilla_vae_model.py
index bc4516d9e953689b58b5277e629b66d275881cfb..3e74b23b9d4969544ef0e560e30cd19e4a983f6b 100644
--- a/video_prediction_tools/model_modules/video_prediction/models/vanilla_vae_model.py
+++ b/video_prediction_tools/model_modules/video_prediction/models/vanilla_vae_model.py
@@ -3,7 +3,7 @@ __email__ = "b.gong@fz-juelich.de"
 __author__ = "Bing Gong"
 __date__ = "2020-09-01"
 
-from model_helpers import set_and_check_pred_frames
+from model_modules.video_prediction.models.model_helpers import set_and_check_pred_frames
 import tensorflow as tf
 from model_modules.video_prediction.layers import layer_def as ld
 from tensorflow.contrib.training import HParams