diff --git a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
index 0d1ef6f6519e6b7a9feb4029516ffa58f638ed00..3c4f049abe6d922543c0094a9ebc64e2ee0ed9d0 100644
--- a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
+++ b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
@@ -1,10 +1,10 @@
 #!/bin/bash
 set -e
 
-export MODULE_NAME=nnunet_for_pytorch
-export MODULE_VERSION=21.11.0
+export MODEL_NAME=nnunet_for_pytorch
+export MODLE_VERSION=21.11.0
 export WORK_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet
-export CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODULE_NAME}_${MODULE_VERSION}.sif
+export CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODEL_NAME}_${MODLE_VERSION}.sif
 export GPU_RESERVATION=nodeimage
 mkdir -p $WORK_DIR/sbatch_out $WORK_DIR/sbatch_scripts $WORK_DIR/results