diff --git a/scripts/benchmark_single_node.sbatch b/scripts/benchmark_single_node.sbatch
index a4c124bf03a58ee175a9c2203e775d008b431f01..ee25c40a6e33a73e67897988860cc6d6c67f1173 100644
--- a/scripts/benchmark_single_node.sbatch
+++ b/scripts/benchmark_single_node.sbatch
@@ -1,8 +1,9 @@
 #!/bin/bash
+sbatch <<EOT
 
 #SBATCH -A nsc
 #SBATCH --nodes=1
-#SBATCH --gres=gpu:8
+#SBATCH --gres=gpu:${3}
 #SBATCH --time=0-0:10:00
 #SBATCH --reservation=devel
 
@@ -31,3 +32,4 @@ apptainer exec --nv -B ${PWD}/data:/data -B ${PWD}/results:/results nvidia_nnu-n
 #rm -f results/benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_amp_iteration${5}.json
 #enroot start --rw --mount ${PWD}/data:/data --mount ${PWD}/results:/results nnunet bash -c "cd /workspace/nnunet_pyt && python scripts/benchmark.py --mode train --gpus ${3} --dim ${1} --batch_size ${4} --nodes ${2} --amp --logname='benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_amp_iteration${5}.json'"
 
+EOT
\ No newline at end of file