diff --git a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/build.txt b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/build.txt
index d7f9057085fe7d6e6215d8bbbb1098e9c38d23d0..b2b105102f28b165a7bf5b43e563100d1e53436a 100644
--- a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/build.txt
+++ b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/build.txt
@@ -14,7 +14,7 @@ apptainer build $CONTAINER_DIR $DEF_DIR
 MODULE_NAME=nnunet_for_pytorch
 MODULE_VERSION=latest
 CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODULE_NAME}_${MODULE_VERSION}.sif
-DEF_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/PyTorch/Segmentation/nnUNet/${MODULE_NAME}_${MODULE_VERSION}.def
+DEF_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/${MODULE_NAME}_${MODULE_VERSION}.def
 
 apptainer build $CONTAINER_DIR $DEF_DIR
 
diff --git a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch_latest.def b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch_latest.def
index fabd296bf34d493813b7f20947dd70d3a994b533..c98805d862b9adfed8aba56ef2e659a4046fdf22 100644
--- a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch_latest.def
+++ b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch_latest.def
@@ -14,7 +14,7 @@ VERSION=latest
 mkdir /workspace/nnunet_pyt 
 cd /workspace/nnunet_pyt
 git clone https://github.com/NVIDIA/DeepLearningExamples.git 
-mv DeepLearningExamples/PyTorch/Segmentation/nnUNet ./
+mv DeepLearningExamples/PyTorch/Segmentation/nnUNet/* ./
 rm -rf DeepLearningExamples
 
 pip install --disable-pip-version-check -r requirements.txt