diff --git a/docker-compose.yml b/docker-compose.yml
index f5b6a68..c46f0dd 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -48,6 +48,7 @@ services:
       - POSTGRES_USER=${POSTGRES_USER}
       - POSTGRES_PASSWORD=${POSTGRES_PASSWORD}
       - POSTGRES_DB=${POSTGRES_DB}
+      - POSTGRES_HOST=postgres
     restart: unless-stopped
     depends_on:
       postgres:
@@ -69,10 +70,10 @@ services:
       - mlflow-data:/mlflow
     command: >
       mlflow server
-      --backend-store-uri /mlflow
-      --default-artifact-root /mlflow/artifacts
-      --host 0.0.0.0
-      --port 5000
+        --backend-store-uri "postgresql://${POSTGRES_USER}:${POSTGRES_PASSWORD}@postgres:5432/${POSTGRES_DB}"
+        --default-artifact-root ./mlruns/artifacts
+        --host 0.0.0.0
+        --port 5000
     restart: unless-stopped
 
   postgres: