diff --git a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
index 2397e4f633efcb25507881860a344a048d02270d..89678627962b0e46e3635501c5c4483ef5d9f607 100644
--- a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
+++ b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
@@ -2,13 +2,13 @@
 set -e
 
 export MODEL_NAME=nnunet_for_pytorch
-export MODLE_VERSION=21.11.0
+export MODLE_VERSION=latest
 export WORK_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet
 export CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODEL_NAME}_${MODLE_VERSION}.sif
 export GPU_RESERVATION=nodeimage
 export TIME_RESERVATIOn=0-0:20:00
-mkdir -p $WORK_DIR/sbatch_out $WORK_DIR/sbatch_scripts $WORK_DIR/results
 
+mkdir -p $WORK_DIR/sbatch_out $WORK_DIR/sbatch_scripts $WORK_DIR/results
 benchmark_modes=("train" "predict")
 # node_types=("thin" "fat")
 node_types=("thin")