diff --git a/mcp_server/docker/docker-compose-falkordb.yml b/mcp_server/docker/docker-compose-falkordb.yml
index ee25b317..99ee3f01 100644
--- a/mcp_server/docker/docker-compose-falkordb.yml
+++ b/mcp_server/docker/docker-compose-falkordb.yml
@@ -51,7 +51,7 @@ services:
       - ../config/config-docker-falkordb.yaml:/app/config/config.yaml:ro
     ports:
       - "8000:8000" # Expose the MCP server via HTTP transport
-    command: ["uv", "run", "src/graphiti_mcp_server.py", "--transport", "http", "--config", "/app/config/config.yaml"]
+    command: ["uv", "run", "main.py"]
 
 volumes:
   falkordb_data:
diff --git a/mcp_server/docker/docker-compose-neo4j.yml b/mcp_server/docker/docker-compose-neo4j.yml
index 2d8a976e..0e3b06fc 100644
--- a/mcp_server/docker/docker-compose-neo4j.yml
+++ b/mcp_server/docker/docker-compose-neo4j.yml
@@ -57,7 +57,7 @@ services:
       - ../config/config-docker-neo4j.yaml:/app/config/config.yaml:ro
     ports:
       - "8000:8000" # Expose the MCP server via HTTP transport
-    command: ["uv", "run", "src/graphiti_mcp_server.py", "--transport", "http", "--config", "/app/config/config.yaml"]
+    command: ["uv", "run", "main.py"]
 
 volumes:
   neo4j_data:
diff --git a/mcp_server/docker/docker-compose.yml b/mcp_server/docker/docker-compose.yml
index 1b3e4cb3..0ee68451 100644
--- a/mcp_server/docker/docker-compose.yml
+++ b/mcp_server/docker/docker-compose.yml
@@ -34,7 +34,7 @@ services:
       - kuzu_data:/data
     ports:
       - "8000:8000" # Expose the MCP server via HTTP transport
-    command: ["uv", "run", "src/graphiti_mcp_server.py", "--transport", "http", "--config", "/app/config/config.yaml"]
+    command: ["uv", "run", "main.py"]
 
 # Volume for persistent KuzuDB storage
 volumes: