diff --git a/video_prediction_tools/model_modules/video_prediction/__init__.py b/video_prediction_tools/model_modules/video_prediction/__init__.py
index 3089b251bbbcbe086a03a4ea63571ce9427b2cb9..aaa5b0f7b3b0a3ef56fabf2ec9d38672df79b99b 100644
--- a/video_prediction_tools/model_modules/video_prediction/__init__.py
+++ b/video_prediction_tools/model_modules/video_prediction/__init__.py
@@ -1,3 +1,3 @@
 from . import losses
 from . import metrics
-from . import ops
+
diff --git a/video_prediction_tools/model_modules/video_prediction/models/test_model.py b/video_prediction_tools/model_modules/video_prediction/models/test_model.py
index f63689877b6bc6d5449cf43ead4c68aa9934eec6..2ae93bc8888e0249164896f94a47678e2f485376 100644
--- a/video_prediction_tools/model_modules/video_prediction/models/test_model.py
+++ b/video_prediction_tools/model_modules/video_prediction/models/test_model.py
@@ -13,12 +13,6 @@ from collections import OrderedDict
 import numpy as np
 import tensorflow as tf
 from tensorflow.python.util import nest
-from model_modules.video_prediction import ops, flow_ops
-from model_modules.video_prediction.models import BaseVideoPredictionModel
-from model_modules.video_prediction.models import networks
-from model_modules.video_prediction.ops import dense, pad2d, conv2d, flatten, tile_concat
-from model_modules.video_prediction.rnn_ops import BasicConv2DLSTMCell, Conv2DGRUCell
-from model_modules.video_prediction.utils import tf_utils
 from datetime import datetime
 from pathlib import Path
 from model_modules.video_prediction.layers import layer_def as ld
diff --git a/video_prediction_tools/utils/runscript_generator/config_preprocess_step2.py b/video_prediction_tools/utils/runscript_generator/config_preprocess_step2.py
index bf433cb1957fd98877328d5900d6deb4e97aa6a2..a2eb48fede7948809f1dcfcc933b384a821eee4d 100755
--- a/video_prediction_tools/utils/runscript_generator/config_preprocess_step2.py
+++ b/video_prediction_tools/utils/runscript_generator/config_preprocess_step2.py
@@ -10,7 +10,7 @@ __date__ = "2021-01-28"
 
 # import modules
 import os, glob
-from data_preprocess.dataset_options import known_datasets
+from data_extraction.dataset_options import known_datasets
 from runscript_generator.config_utils import Config_runscript_base    # import parent class
 
 class Config_Preprocess2(Config_runscript_base):