diff --git a/video_prediction_tools/env_setup/install_venv_container.sh b/video_prediction_tools/env_setup/install_venv_container.sh
index 33f3929b5a51224960ece3efe68d6717ce06b660..f0d53cdb495d5df6dffd30b79a19f53e1a0b2e98 100755
--- a/video_prediction_tools/env_setup/install_venv_container.sh
+++ b/video_prediction_tools/env_setup/install_venv_container.sh
@@ -13,6 +13,7 @@ BASE_DIR="$(pwd)"
 VENV_DIR=$1
 VENV_NAME="$(basename "${VENV_DIR}")"
 VENV_BASE="$(dirname "${VENV_DIR}")"
+WORKING_DIR="$(dirname "${VENV_BASE}")"
 VENV_REQ=${BASE_DIR}/requirements.txt
 
 # sanity checks