diff --git a/video_prediction_tools/main_scripts/main_data_extraction.py b/video_prediction_tools/main_scripts/main_data_extraction.py
index 494540eedf9553271ff3a3e9e901a34c0d1236fc..c9ec40d4860ab4797013cbf18b512c3c8437c5db 100644
--- a/video_prediction_tools/main_scripts/main_data_extraction.py
+++ b/video_prediction_tools/main_scripts/main_data_extraction.py
@@ -13,7 +13,7 @@ import logging
 import time
 from utils.external_function import directory_scanner
 from utils.external_function import load_distributor
-from data_preprocess.prepare_era5_data import *
+from data_preprocess.era5_data_extraction import *
 # How to Run it!
 # mpirun -np 6 python mpi_stager_v2.py
 import os
diff --git a/video_prediction_tools/main_scripts/main_preprocess_data_step1.py b/video_prediction_tools/main_scripts/main_preprocess_data_step1.py
index 0c675df6b6295b053aa9ebb97bd62f768b31bcfe..e42cfb088f70fd9cbcb5e0649285358e229a84a6 100755
--- a/video_prediction_tools/main_scripts/main_preprocess_data_step1.py
+++ b/video_prediction_tools/main_scripts/main_preprocess_data_step1.py
@@ -17,7 +17,7 @@ import time
 import argparse
 from utils.external_function import directory_scanner
 from utils.external_function import load_distributor
-from data_preprocess.process_netCDF_v2 import *  
+from data_preprocess.preprocess_data_step1 import *  
 from metadata import MetaData
 from netcdf_datahandling import GeoSubdomain
 import json