diff --git a/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh b/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh
index d46d924628e7f2b720ef99c90d3d8effb3cf6af9..0e3718d454691f192f6a7f0161f089d206c7f699 100644
--- a/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh
+++ b/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh
@@ -3,8 +3,9 @@
 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/benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_iteration${5}.sbatch
+SBATCH_DIR=$WORK_DIR/sbatch_scripts/benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_iteration${5}.sbatch
 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
 
 cat <<EOT >  $SBATCH_DIR
 #!/bin/bash
@@ -21,7 +22,7 @@ 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
 
-mkdir -p $WORK_DIR/sbatch_out 
+
 
 cd $WORK_DIR