diff --git a/assets/training/model_evaluation/environment/context/Dockerfile b/assets/training/model_evaluation/environment/context/Dockerfile
index 8b8429aa48..d1db377bfc 100644
--- a/assets/training/model_evaluation/environment/context/Dockerfile
+++ b/assets/training/model_evaluation/environment/context/Dockerfile
@@ -22,3 +22,6 @@ ENV PATH=$CONDA_PREFIX/bin:$PATH
 
 # This is needed for mpi to locate libpython
 ENV LD_LIBRARY_PATH $CONDA_PREFIX/lib:$LD_LIBRARY_PATH
+
+# Temp fix for env vuln
+RUN pip install mlflow==2.16.0  
diff --git a/assets/training/model_evaluation/environment/spec.yaml b/assets/training/model_evaluation/environment/spec.yaml
index 6fa4a3e07a..e0ab2c6551 100644
--- a/assets/training/model_evaluation/environment/spec.yaml
+++ b/assets/training/model_evaluation/environment/spec.yaml
@@ -13,4 +13,4 @@ os_type: linux
 tags:
   OS: Ubuntu20.04
   Evaluation: ""
-  MLFlow: "2.2.1"
+  MLFlow: "2.16.0"