diff --git a/video_prediction_savp/HPC_scripts/data_extraction_era5_template.sh b/video_prediction_savp/HPC_scripts/data_extraction_era5_template.sh
index 68e1c97c2575e6002694ca895f618a424f22d822..e00e3591f09e6c7feca29fc3dbf32185cab0a64b 100644
--- a/video_prediction_savp/HPC_scripts/data_extraction_era5_template.sh
+++ b/video_prediction_savp/HPC_scripts/data_extraction_era5_template.sh
@@ -42,7 +42,7 @@ dest_dir="/p/scratch/deepacf/video_prediction_shared_folder/extractedData/"
 year="2010"
 
 # Run data extraction
-srun python ../../workflow_parallel_frame_prediction/DataExtraction/mpi_stager_v2.py --source_dir ${source_dir}/${year}/ --destination_dir ${dest_dir}/${year}/
+srun python ../main_scripts/main_data_extraction.py  --source_dir ${source_dir}/${year}/ --destination_dir ${dest_dir}/${year}/