diff --git a/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh b/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh
index 0e3718d454691f192f6a7f0161f089d206c7f699..95071f9a6b5f73c5c2be91e3a84b239c31bd15ed 100644
--- a/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh
+++ b/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh
@@ -4,6 +4,7 @@ MODULE_NAME=nnunet_for_pytorch
 MODULE_VERSION=21.11.0
 WORK_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/PyTorch/Segmentation/nnUNet
 SBATCH_DIR=$WORK_DIR/sbatch_scripts/benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_iteration${5}.sbatch
+SBATCH_OUT_DIR=$WORK_DIR/sbatch_out/benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_iteration${5}.out
 LOG_DIR=benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_amp_iteration${5}.json
 mkdir -p $WORK_DIR/sbatch_out $WORK_DIR/sbatch_scripts
 
@@ -14,7 +15,7 @@ cat <<EOT >  $SBATCH_DIR
 #SBATCH --nodes=${2}
 #SBATCH --gpus=${3}
 #SBATCH --time=0-0:10:00
-#SBATCH --output=sbatch_out/benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_iteration${5}.out
+#SBATCH --output=$SBATCH_OUT_DIR
 #SBATCH --reservation=devel
 
 MODULE_NAME=nnunet_for_pytorch
@@ -22,8 +23,6 @@ MODULE_VERSION=21.11.0
 WORK_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/PyTorch/Segmentation/nnUNet
 CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODULE_NAME}_${MODULE_VERSION}.sif
 
-
-
 cd $WORK_DIR
 
 rm -f results/$LOG_DIR