diff --git a/cognee/api/v1/ui/ui.py b/cognee/api/v1/ui/ui.py
index d0f109393..2a5746f8c 100644
--- a/cognee/api/v1/ui/ui.py
+++ b/cognee/api/v1/ui/ui.py
@@ -503,7 +503,7 @@ def start_ui(
     if start_mcp:
         logger.info("Starting Cognee MCP server with Docker...")
         try:
-            image = "cognee/cognee-mcp:main"
+            image = "cognee/cognee-mcp:feature-standalone-mcp"  # TODO: change to main right before merging into main
             subprocess.run(["docker", "pull", image], check=True)
 
             import uuid
@@ -538,7 +538,9 @@ def start_ui(
                 env_file = os.path.join(cwd, ".env")
                 docker_cmd.extend(["--env-file", env_file])
 
-            docker_cmd.append("cognee/cognee-mcp:main")
+            docker_cmd.append(
+                "cognee/cognee-mcp:feature-standalone-mcp"
+            )  # TODO: change to main right before merging into main
 
             mcp_process = subprocess.Popen(
                 docker_cmd,