diff --git a/kubernetes/apps/datasci/mlflow/app/helmrelease.yaml b/kubernetes/apps/datasci/mlflow/app/helmrelease.yaml
index 436af1c14..646f6a9eb 100755
--- a/kubernetes/apps/datasci/mlflow/app/helmrelease.yaml
+++ b/kubernetes/apps/datasci/mlflow/app/helmrelease.yaml
@@ -90,6 +90,8 @@ spec:
               MLFLOW_SERVE_ARTIFACTS: true
               MLFLOW_HOST: 0.0.0.0
               MLFLOW_PORT: &port 5000
+              # To connect the MLflow AI Gateway with the MLflow Tracking Server
+              MLFLOW_DEPLOYMENTS_TARGET: http://mlflow-gateway.datasci.svc.cluster.local:5001
             # envFrom:
             #   - secretRef:
             #       name: database-mlflow
@@ -180,7 +182,7 @@ spec:
           http:
             port: *port
       gateway:
-        controller: mlflow
+        controller: gateway
         ports:
           http:
             port: *gport
@@ -207,7 +209,7 @@ spec:
           nginx.ingress.kubernetes.io/whitelist-source-range: |
             10.0.0.0/8,172.16.0.0/12,192.168.0.0/16
         hosts:
-          - host: &host mlflow-gateway.${SECRET_DOMAIN}
+          - host: &gateway mlflow-gateway.${SECRET_DOMAIN}
             paths:
               - path: /
                 pathType: Prefix
@@ -216,7 +218,7 @@ spec:
                   port: http
         tls:
           - hosts:
-              - *host
+              - *gateway
     persistence:
       data:
         enabled: true