diff --git a/.github/workflows/vector_db_tests.yml b/.github/workflows/vector_db_tests.yml
index cc2071b72..6e9e34493 100644
--- a/.github/workflows/vector_db_tests.yml
+++ b/.github/workflows/vector_db_tests.yml
@@ -149,7 +149,7 @@ jobs:
           VECTOR_DB_PORT: 5432
           VECTOR_DB_USERNAME: cognee
           VECTOR_DB_PASSWORD: ${{ secrets.POSTGRES_PASSWORD }}
-          VECTOR_DATASET_DATABASE_HANDLER: pgvector_local
+          VECTOR_DATASET_DATABASE_HANDLER: pgvector
         run: uv run python ./cognee/tests/test_permissions.py
 
   run-lancedb-tests:
diff --git a/cognee/infrastructure/databases/dataset_database_handler/supported_dataset_database_handlers.py b/cognee/infrastructure/databases/dataset_database_handler/supported_dataset_database_handlers.py
index 5ba1a91a1..1bac5c4ef 100644
--- a/cognee/infrastructure/databases/dataset_database_handler/supported_dataset_database_handlers.py
+++ b/cognee/infrastructure/databases/dataset_database_handler/supported_dataset_database_handlers.py
@@ -17,7 +17,7 @@ supported_dataset_database_handlers = {
         "handler_provider": "neo4j",
     },
     "lancedb": {"handler_instance": LanceDBDatasetDatabaseHandler, "handler_provider": "lancedb"},
-    "pgvector_local": {
+    "pgvector": {
         "handler_instance": PGVectorDatasetDatabaseHandler,
         "handler_provider": "pgvector",
     },
diff --git a/cognee/infrastructure/databases/vector/pgvector/PGVectorDatasetDatabaseHandler.py b/cognee/infrastructure/databases/vector/pgvector/PGVectorDatasetDatabaseHandler.py
index 682c5635c..22e8bb6d0 100644
--- a/cognee/infrastructure/databases/vector/pgvector/PGVectorDatasetDatabaseHandler.py
+++ b/cognee/infrastructure/databases/vector/pgvector/PGVectorDatasetDatabaseHandler.py
@@ -34,7 +34,7 @@ class PGVectorDatasetDatabaseHandler(DatasetDatabaseHandlerInterface):
                 "password": vector_config.vector_db_password,
                 "host": vector_config.vector_db_host,
             },
-            "vector_dataset_database_handler": "pgvector_local",
+            "vector_dataset_database_handler": "pgvector",
         }
 
         from .create_db_and_tables import create_pg_database
@@ -59,6 +59,12 @@ class PGVectorDatasetDatabaseHandler(DatasetDatabaseHandlerInterface):
 
         return new_vector_config
 
+    @classmethod
+    async def resolve_dataset_connection_info(
+        cls, dataset_database: DatasetDatabase
+    ) -> DatasetDatabase:
+        return dataset_database
+
     @classmethod
     async def delete_dataset(cls, dataset_database: DatasetDatabase):
         vector_engine = create_vector_engine(