diff --git a/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh b/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh
index c8b17ae37ffd042aff8c889eea2b970b6c7a797c..c837ccde74ff98f78ca38c55de9fcd97514b9fd2 100644
--- a/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh
+++ b/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh
@@ -1,7 +1,7 @@
 #!/bin/bash
 set -e
 
-export MODEL_NAME=nnunet_for_pytorch
+export MODEL_NAME=U-Net3D
 export WORK_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/MLPerf/training/image_segmentation/pytorch
 export CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODEL_NAME}.sif
 export GPU_RESERVATION=