diff --git a/continuous-integration/code-generation/generate_datamodels.sh b/continuous-integration/code-generation/generate_datamodels.sh
index 43ca002d..d5dcfd76 100755
--- a/continuous-integration/code-generation/generate_datamodels.sh
+++ b/continuous-integration/code-generation/generate_datamodels.sh
@@ -8,6 +8,11 @@
 SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
 ROOT_DIR=$(realpath "$SCRIPT_DIR/../..")
 
+if [ -z "$VIRTUAL_ENV" ]; then
+    source $SCRIPT_DIR/../linux/venv.sh || exit $?
+    activate_venv || exit $?
+fi
+
 python3 -m pip install --requirement "$SCRIPT_DIR/../python-requirements/lint.txt" || exit $?
 
 python3 "$ROOT_DIR/continuous-integration/code-generation/datamodel_frontend_generator.py" \