diff --git a/scripts/benchmark_sbatch_submit.sh b/scripts/benchmark_sbatch_submit.sh
index f3459210265b1b5e0436ceccabe7caaba4fe5b82..a75647014ad8b785d5e658cc05de8500571c4c1e 100644
--- a/scripts/benchmark_sbatch_submit.sh
+++ b/scripts/benchmark_sbatch_submit.sh
@@ -8,6 +8,7 @@ if [ $NUM_NODES -eq 1 ]; then
         for gpus in {1,8}; do
             for batch_size in {64,128}; do
                 for iteration in {1..1}; 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}
@@ -23,11 +24,12 @@ else
         for gpus in {8,8}; do
             for batch_size in {64,128}; do
                 for iteration in {1..1}; do
-                    batch_size=$BATCH_SIZE
+
                     # 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}
 
                     sleep 1 # pause to be kind to the scheduler
+
                 done
             done
         done