diff --git a/video_prediction_tools/utils/runscript_generator/config_training.py b/video_prediction_tools/utils/runscript_generator/config_training.py
index 38d3ade765f1ccce84b241d824b7d63c13eb6447..1248a196a0ebb50234d9a053a7cc4c159b5664b4 100755
--- a/video_prediction_tools/utils/runscript_generator/config_training.py
+++ b/video_prediction_tools/utils/runscript_generator/config_training.py
@@ -17,7 +17,7 @@ import subprocess as sp
 from pathlib import Path
 from model_modules.model_architectures import known_models
 
-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