diff --git a/build.txt b/build.txt
index 4767274e1f448809ba8b70b6066972581a63c96d..26d71ae0432e7590885921371ac9e83ae4a36b08 100644
--- a/build.txt
+++ b/build.txt
@@ -1,4 +1,4 @@
-apptainer build containers/nvidia_pytorch:21.11-py3.sif  docker://nvcr.io/nvidia/pytorch:21.11-py3
+apptainer build containers/nvidia_pytorch_21.11-py3.sif  docker://nvcr.io/nvidia/pytorch:21.11-py3
 
 
 MODULE_NAME=nnunet_for_pytorch
diff --git a/nnunet_for_pytorch.def b/nnunet_for_pytorch.def
index 1c015b62a89d4e15a804737e7d1aa21ff68b925d..f5803343dee7840aa6e710b1a652b26938dc7994 100644
--- a/nnunet_for_pytorch.def
+++ b/nnunet_for_pytorch.def
@@ -1,5 +1,5 @@
 Bootstrap: localimage
-From: /proj/nsc_testing/xuan/containers/nvidia_pytorch:21.11-py3.sif
+From: /proj/nsc_testing/xuan/containers/nvidia_pytorch_21.11-py3.sif
 
 
 %environment