diff --git a/build.txt b/build.txt
index b15d32bd5625cac9d5b1d3839d03f47825c56c93..4767274e1f448809ba8b70b6066972581a63c96d 100644
--- a/build.txt
+++ b/build.txt
@@ -4,5 +4,5 @@ apptainer build containers/nvidia_pytorch:21.11-py3.sif  docker://nvcr.io/nvidia
 MODULE_NAME=nnunet_for_pytorch
 MODULE_VERSION=21.11.0
 gotesting
-apptainer build ./containers/${MODULE_NAME}_${MODULE_VERSION}.sif ./berzelius-container-modules/AlphaFold/${MODULE_VERSION}_mamba.def
+apptainer build ./containers/${MODULE_NAME}_${MODULE_VERSION}.sif ./berzelius-benchmarks/${MODULE_NAME}.def