diff --git a/README.md b/README.md
index 92841f2620fde2f3c501bae62c2e53e0eb61a749..79574e00302e8d854b2c417c1de7da1f0c672691 100644
--- a/README.md
+++ b/README.md
@@ -82,7 +82,6 @@ bash srtips/copy_data_for_benchmark.sh
 - Run the script.   
 You need to modify the script for e.g. the name of your reservation, number of nodes, batch_size, etc. Also, choose either singularity or enroot.
 ```
-mkdir sbatch_out
 bash srtips/benchmark_sbatch_submit.sh
 ```
 
diff --git a/scripts/benchmark_sbatch_submit.sh b/scripts/benchmark_sbatch_submit.sh
index a2aa7b627081079846650f24a820d17d92489991..d9886db94e6d109946a99025f9771cbe91a97131 100644
--- a/scripts/benchmark_sbatch_submit.sh
+++ b/scripts/benchmark_sbatch_submit.sh
@@ -1,22 +1,25 @@
 cd /proj/nsc_testing/xuan/Berzelius-nnU-Net-Benchmark
 
-dim=2
+mkdir -p sbatch_out
 
-nodes=1
-for gpus in {1..8}; do
-    for batch_size in 128; do
-        for iteration in {1..100}; do  
+dim=2
+for nodes in {1..1}; do
+    for gpus in {1..8}; do
+        for batch_size in 128; do
+            for iteration in {1..100}; 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}     
+                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 benchmark_single_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
 
-            sleep 1 # pause to be kind to the scheduler
+                sleep 1 # pause to be kind to the scheduler
 
+            done
         done
     done
 done
 
+
 for nodes in {2..8}; do
     for gpus in {8}; do
         for batch_size in 128; do