diff --git a/cognee-mcp/pyproject.toml b/cognee-mcp/pyproject.toml
index d6f5b986a..7820bfe81 100644
--- a/cognee-mcp/pyproject.toml
+++ b/cognee-mcp/pyproject.toml
@@ -7,9 +7,8 @@ requires-python = ">=3.10"
 
 dependencies = [
     # For local cognee repo usage remove comment bellow and add absolute path to cognee. Then run `uv sync --reinstall` in the mcp folder on local cognee changes.
-    "cognee[postgres,codegraph,gemini,huggingface,docs,neo4j] @ file:/Users/igorilic/Desktop/cognee/dist/cognee-0.5.0.dev0-py3-none-any.whl",
-    # TODO: Remove gemini from optional dependecnies for new Cognee version after 0.3.4
-    #"cognee[postgres,docs,neo4j]==0.5.0",
+    #"cognee[postgres,codegraph,huggingface,docs,neo4j] @ file:/Users/igorilic/Desktop/cognee",
+    "cognee[postgres,docs,neo4j]==0.5.0",
     "fastmcp>=2.10.0,<3.0.0",
     "mcp>=1.12.0,<2.0.0",
     "uv>=0.6.3,<1.0.0",
diff --git a/cognee/run_migrations.py b/cognee/run_migrations.py
index 11347d389..c08c0fc41 100644
--- a/cognee/run_migrations.py
+++ b/cognee/run_migrations.py
@@ -44,3 +44,5 @@ async def run_migrations():
         migration_output = migration_result.stderr + migration_result.stdout
         print(f"Migration failed with unexpected error: {migration_output}")
         sys.exit(1)
+
+    print("Migration completed successfully.")