diff --git a/Scripts/grid_search.sh b/Scripts/grid_search.sh
index 24a98096952552a12ac35e38e689c13430c8bf35..54cfff5abf9ba135b551e8a49e3ba8a8e4068da2 100644
--- a/Scripts/grid_search.sh
+++ b/Scripts/grid_search.sh
@@ -67,9 +67,7 @@ for predictand in ${PREDICTAND[@]}; do
                       	sed -i "s/'weight_decay':.*/'weight_decay': $lambda/" ./climax/main/config.py
 
                        	# run downscaling
-                        # python climax/main/downscale.py
-                        python climax/main/downscale_train.py
-                        python climax/main/downscale_infer.py
+                        python climax/main/downscale.py
                     done
                 done
             else
@@ -82,9 +80,7 @@ for predictand in ${PREDICTAND[@]}; do
                     sed -i "s/'weight_decay':.*/'weight_decay': $lambda/" ./climax/main/config.py
 
                     # run downscaling
-                    # python climax/main/downscale.py
-                    python climax/main/downscale_train.py
-                    python climax/main/downscale_infer.py
+                    python climax/main/downscale.py
                 done
             fi
         done