diff --git a/video_prediction_tools/main_scripts/main_visualize_postprocess.py b/video_prediction_tools/main_scripts/main_visualize_postprocess.py
index 0c6504115baa11c6764e1d3933b493c7f4acfe8c..9d008cf0e756b6842efe29aa8a4c343818abfd8d 100644
--- a/video_prediction_tools/main_scripts/main_visualize_postprocess.py
+++ b/video_prediction_tools/main_scripts/main_visualize_postprocess.py
@@ -19,7 +19,8 @@ import json
 from typing import Union, List
 # own modules
 from normalization import Norm_data
-from general_utils import get_era5_varatts, check_dir
+from netcdf_datahandling import get_era5_varatts
+from general_utils import check_dir
 from metadata import MetaData as MetaData
 from main_scripts.main_train_models import *
 from data_preprocess.preprocess_data_step2 import *