diff --git a/infrastructure/helm-charts/mlflow/Dockerfile b/infrastructure/helm-charts/mlflow/Dockerfile
index cfd7dfa..fd5de53 100644
--- a/infrastructure/helm-charts/mlflow/Dockerfile
+++ b/infrastructure/helm-charts/mlflow/Dockerfile
@@ -6,11 +6,11 @@ ENV BACKEND_STORE_URI=$TRACKING_STORE
 ARG ARTIFACT_STORE="/mnt/mlflow_data"
 ENV DEFAULT_ARTIFACT_ROOT=$ARTIFACT_STORE
 
-RUN pip install -U mlflow==1.2.0
+RUN pip install -U mlflow==1.5.0
 
 EXPOSE 5000
 
-CMD mlflow server \
+CMD mlflow ui \
     --backend-store-uri $BACKEND_STORE_URI \
     --default-artifact-root $DEFAULT_ARTIFACT_ROOT \
     --host 0.0.0.0 --port 5000