diff --git a/video_prediction_tools/env_setup/wrapper_container.sh b/video_prediction_tools/env_setup/wrapper_container.sh
index 4a51d0d1812de20cd391d4c0f9bc4623de44403a..b4c2f3f411d3306b3aadea6b9e4ae66a6a8f93fb 100755
--- a/video_prediction_tools/env_setup/wrapper_container.sh
+++ b/video_prediction_tools/env_setup/wrapper_container.sh
@@ -23,7 +23,7 @@ fi
 unset PYTHONPATH
 export PYTHONPATH=${VENV_DIR}/lib/python3.8/site-packages/:$PYTHONPATH
 # ... dist-packages from container singularity...
-#export PYTHONPATH=/usr/local/lib/python3.8/dist-packages:$PYTHONPATH
+export PYTHONPATH=/usr/local/lib/python3.8/dist-packages:$PYTHONPATH
 # ... and modules from this project
 export PYTHONPATH=${WORKING_DIR}:$PYTHONPATH
 export PYTHONPATH=${WORKING_DIR}/utils:$PYTHONPATH