diff --git a/scripts/benchmark_sbatch_submit.sh b/scripts/benchmark_sbatch_submit.sh
index 355e01e8523ee07735692b8b659101f6a2d6267c..da5b934c3f2d3e3969b77ffca7eab0b5583f3097 100644
--- a/scripts/benchmark_sbatch_submit.sh
+++ b/scripts/benchmark_sbatch_submit.sh
@@ -8,14 +8,14 @@ NUM_GPUS=$2
 NUM_ITERATIONS=$3
 BATCH_SIZE=$4
 if [ $NUM_NODES -eq 1 ]; then
-    for nodes in {1..1}; do
-        for gpus in {1..$NUM_GPUS}; do
-            for batch_size in $BATCH_SIZE; do
-                for iteration in {1..$NUM_ITERATIONS}; do
+    for nodes in {1..$((NUM_NODES))}; do
+        for gpus in {1..$((NUM_GPUS))}; do
+            for batch_size in $((BATCH_SIZE)); do
+                for iteration in {1..$((NUM_ITERATIONS))}; do
 
                     echo dim ${dim}, nodes ${nodes}, gpus ${gpus}, batch_size ${batch_size}, iteration ${iteration}
                     # For single node
-                    sbatch -o sbatch_out/benchmark_dim${dim}_nodes${nodes}_gpus${gpus}_batchsize${batch_size}_iteration${iteration}.out scripts/benchmark_single_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
+                    #sbatch -o sbatch_out/benchmark_dim${dim}_nodes${nodes}_gpus${gpus}_batchsize${batch_size}_iteration${iteration}.out scripts/benchmark_single_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
 
                     sleep 1 # pause to be kind to the scheduler
 
@@ -24,10 +24,10 @@ if [ $NUM_NODES -eq 1 ]; then
         done
     done
 else
-    for nodes in {2..$NUM_NODES}; do
-        for gpus in {$NUM_GPUS}; do
-            for batch_size in $BATCH_SIZE; do
-                for iteration in {1..$NUM_ITERATIONS}; do
+    for nodes in {2..$((NUM_NODES))}; do
+        for gpus in {$((NUM_GPUS))}; do
+            for batch_size in $((BATCH_SIZE)); do
+                for iteration in {1..$((NUM_ITERATIONS))}; do
 
                     # For multi node
                     sbatch -o sbatch_out/benchmark_dim${dim}_nodes${nodes}_gpus${gpus}_batchsize${batch_size}_iteration${iteration}.out scripts/benchmark_multi_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}