diff --git a/container.def b/container.def
index a8c2a8b30a87146166f1a69569019bedbb9c9780..ee33b7e6ea7a65a7a18a0c53e51c049e9792a8f7 100644
--- a/container.def
+++ b/container.def
@@ -27,8 +27,8 @@ From: nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
     /opt/conda/bin/conda init zsh
     /opt/conda/bin/conda init tcsh
 
-    # Install Python packages from requirements.yaml into the base environment
-    /opt/conda/bin/conda env update -n base --file /opt/env/requirements.yaml --prune
+    # Install Python packages from requirements.yml into the base environment
+    /opt/conda/bin/conda env update -n base --file /opt/env/requirements.yml --prune
 
 %environment
     # Set some environment vars