diff --git a/cognee/infrastructure/databases/dataset_database_handler/dataset_database_handler_interface.py b/cognee/infrastructure/databases/dataset_database_handler/dataset_database_handler_interface.py
index 01ee46c48..1811feb12 100644
--- a/cognee/infrastructure/databases/dataset_database_handler/dataset_database_handler_interface.py
+++ b/cognee/infrastructure/databases/dataset_database_handler/dataset_database_handler_interface.py
@@ -68,7 +68,7 @@ class DatasetDatabaseHandlerInterface(ABC):
 
     @classmethod
     @abstractmethod
-    async def delete_dataset(cls, dataset_id: UUID, user: User) -> None:
+    async def delete_dataset(cls, dataset_id: Optional[UUID], user: Optional[User]) -> None:
         """
         Delete the graph or vector database for the given dataset.
         Function should auto handle deleting of the actual database or send a request to the proper service to delete/mark the database as not needed for the given dataset.
diff --git a/cognee/infrastructure/databases/graph/kuzu/KuzuDatasetDatabaseHandler.py b/cognee/infrastructure/databases/graph/kuzu/KuzuDatasetDatabaseHandler.py
index a2b2da8f4..242249e00 100644
--- a/cognee/infrastructure/databases/graph/kuzu/KuzuDatasetDatabaseHandler.py
+++ b/cognee/infrastructure/databases/graph/kuzu/KuzuDatasetDatabaseHandler.py
@@ -1,6 +1,3 @@
-import os
-import asyncio
-import requests
 from uuid import UUID
 from typing import Optional