diff --git a/PyTorch/Segmentation/nnUNet/build.txt b/PyTorch/Segmentation/nnUNet/build.txt
index 47a7948a0170dd57d7ee059b72371e95af8088bd..e3e069b2b607dd399e7435f50bbfca4611fde5f3 100644
--- a/PyTorch/Segmentation/nnUNet/build.txt
+++ b/PyTorch/Segmentation/nnUNet/build.txt
@@ -8,7 +8,7 @@ gotesting
 MODULE_NAME=nnunet_for_pytorch
 MODULE_VERSION=21.11.0
 
-apptainer build ./containers/${MODULE_NAME}_${MODULE_VERSION}.sif ./berzelius-benchmarks/PyTorch/Segmentation/nnUNet/${MODULE_NAME}.def
+apptainer build ./containers/${MODULE_NAME}_${MODULE_VERSION}.sif ./berzelius-benchmarks/PyTorch/Segmentation/nnUNet/${MODULE_NAME}_${MODULE_VERSION}.def
 
 
 WORK_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/PyTorch/Segmentation/nnUNet
diff --git a/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch_21.11.0.def b/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch_21.11.0.def
index 03f7621a7186a0a9733b376c1f93a77aeadcb502..99b03eaf38e88ef6b15462ff8f884de3da36bff9 100644
--- a/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch_21.11.0.def
+++ b/PyTorch/Segmentation/nnUNet/nnunet_for_pytorch_21.11.0.def
@@ -18,19 +18,10 @@ unzip nnunet_for_pytorch_21.11.0.zip
 
 # Fix for https://github.com/NVIDIA/DeepLearningExamples/issues/1113
 #sed 's/pytorch-lightning==1.3.8/pytorch-lightning==1.5.10/g' requirements.txt > temp_file && mv temp_file requirements.txt
-#pip install --disable-pip-version-check -r requirements.txt
-#pip install monai==0.8.0 --no-dependencies
-#pip uninstall -y torchtext
-
-conda install -y nibabel==3.2.1 \
-joblib==1.0.1 \
-pytorch-lightning==1.3.8 \
-scikit-learn==1.0 \
-scikit-image==0.18.3 \
-pynvml==11.0.0 \
-monai==0.8.0
-
-conda remove -y torchtext
+pip install --disable-pip-version-check -r requirements.txt
+pip install monai==0.8.0 --no-dependencies
+pip uninstall -y torchtext torchmetrics
+pip install torchmetrics==0.6.0
 
 curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "awscliv2.zip"
 unzip -qq awscliv2.zip