diff --git a/video_prediction_tools/env_setup/create_env.sh b/video_prediction_tools/env_setup/create_env.sh
index 079007f75b9c42f1139c3f3006f11d4979f51d04..503553fe195d0bddf5b51cea3ffce2fed1f00eed 100755
--- a/video_prediction_tools/env_setup/create_env.sh
+++ b/video_prediction_tools/env_setup/create_env.sh
@@ -54,7 +54,7 @@ ENV_SETUP_DIR=`pwd`
 WORKING_DIR="$(dirname "$ENV_SETUP_DIR")"
 EXE_DIR="$(basename "$ENV_SETUP_DIR")"
 ENV_DIR=${WORKING_DIR}/${ENV_NAME}
-TF_CONTAINER=${WORKING_DIR}/tensorflow_21.09-tf1-py3.sif
+TF_CONTAINER=${WORKING_DIR}/../env_setup/tensorflow_21.09-tf1-py3.sif
 
 ## perform sanity checks
 # correct bool_container if host is Juwels Booster and ensure running singularity