diff --git a/cognee/tasks/graph/extract_graph_from_data.py b/cognee/tasks/graph/extract_graph_from_data.py
index 01d8bb618..433cbe591 100644
--- a/cognee/tasks/graph/extract_graph_from_data.py
+++ b/cognee/tasks/graph/extract_graph_from_data.py
@@ -34,7 +34,6 @@ async def integrate_chunk_graphs(
     existing_edges_map = await retrieve_existing_edges(
         data_chunks,
         chunk_graphs,
-        graph_engine,
     )
 
     graph_nodes, graph_edges = expand_with_nodes_and_edges(
diff --git a/distributed/tasks/save_data_points.py b/distributed/tasks/save_data_points.py
index 794cb36e1..6057e3a3b 100644
--- a/distributed/tasks/save_data_points.py
+++ b/distributed/tasks/save_data_points.py
@@ -4,9 +4,7 @@ from cognee.modules.graph.utils import deduplicate_nodes_and_edges, get_graph_fr
 from distributed.queues import save_data_points_queue
 
 
-async def save_data_points(
-    data_points_and_relationships: tuple[list, list]
-):
+async def save_data_points(data_points_and_relationships: tuple[list, list]):
     data_points = data_points_and_relationships[0]
     data_point_connections = data_points_and_relationships[1]
 
diff --git a/distributed/tasks/summarize_text.py b/distributed/tasks/summarize_text.py
index fde4b19b1..32cd33a47 100644
--- a/distributed/tasks/summarize_text.py
+++ b/distributed/tasks/summarize_text.py
@@ -8,7 +8,8 @@ from cognee.modules.chunking.models.DocumentChunk import DocumentChunk
 
 
 async def summarize_text(
-    data_points_and_relationships: tuple[list[DocumentChunk], list], summarization_model: Type[BaseModel]
+    data_points_and_relationships: tuple[list[DocumentChunk], list],
+    summarization_model: Type[BaseModel],
 ):
     data_chunks = data_points_and_relationships[0]
     edges = data_points_and_relationships[1]