diff --git a/cognee/modules/pipelines/operations/run_tasks_distributed.py b/cognee/modules/pipelines/operations/run_tasks_distributed.py
index 027f1c46c..26e9375b9 100644
--- a/cognee/modules/pipelines/operations/run_tasks_distributed.py
+++ b/cognee/modules/pipelines/operations/run_tasks_distributed.py
@@ -61,8 +61,6 @@ if modal:
         from cognee.infrastructure.databases.relational import get_relational_engine
 
         async with get_relational_engine().get_async_session() as session:
-            from cognee.modules.data.models import Dataset
-
             dataset = await session.get(Dataset, dataset_id)
 
         result = await run_tasks_data_item(