diff --git a/scripts/run_benchmark.sh b/scripts/run_benchmark.sh
index bd012fae5b3a66476699320c8119b619be49947e..4a55f1ab5d28202d8d51318121f2c00534b938d7 100644
--- a/scripts/run_benchmark.sh
+++ b/scripts/run_benchmark.sh
@@ -11,16 +11,14 @@ for nodes in {1..1}; do
 
                 # For single node
                 if [ $nodes -eq 1 ]; then
-                    bash scripts/generate_single_node_job.sh ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
-                    sbatch scripts/benchmark_single_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
+                    bash scripts/run_benchmark_single_node.sh ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
 
                     sleep 1 # pause to be kind to the scheduler
                 else
 
                     # For multi node
                     gpus=8
-                    bash scripts/generate_multi_node_job.sh ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
-                    #sbatch scripts/benchmark_multi_nodebatc ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
+                    bash scripts/run_benchmark_multi_node.sh ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
 
                     sleep 1 # pause to be kind to the scheduler
                 fi
@@ -39,16 +37,14 @@ for nodes in {1..1}; do
 
                 # For single node
                 if [ $nodes -eq 1 ]; then
-                    bash scripts/generate_single_node_job.sh ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
-                    #sbatch scripts/benchmark_single_nodebatc ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
+                    bash scripts/run_benchmark_single_node.sh ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
 
                     sleep 1 # pause to be kind to the scheduler
                 else
 
                     # For multi node
                     gpus=8
-                    bash scripts/generate_multi_node_job.sh ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
-                    #sbatch scripts/benchmark_multi_nodebatc ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
+                    bash scripts/run_benchmark_multi_node.sh ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
 
                     sleep 1 # pause to be kind to the scheduler
                 fi
diff --git a/scripts/run_benchmark_single_node_job.sh b/scripts/run_benchmark_single_node.sh
similarity index 100%
rename from scripts/run_benchmark_single_node_job.sh
rename to scripts/run_benchmark_single_node.sh
diff --git a/scripts/run_bencmark_multi_node_job.sh b/scripts/run_bencmark_multi_node.sh
similarity index 100%
rename from scripts/run_bencmark_multi_node_job.sh
rename to scripts/run_bencmark_multi_node.sh