diff --git a/cognee/tasks/storage/add_data_points.py b/cognee/tasks/storage/add_data_points.py
index 68e6404f8..1c7a4c2bc 100644
--- a/cognee/tasks/storage/add_data_points.py
+++ b/cognee/tasks/storage/add_data_points.py
@@ -11,7 +11,7 @@ from cognee.tasks.storage.exceptions import (
 
 
 async def add_data_points(
-    data_points: List[DataPoint], update_edge_collection=True
+    data_points: List[DataPoint], update_edge_collection: bool = True
 ) -> List[DataPoint]:
     """
     Add a batch of data points to the graph database by extracting nodes and edges,
@@ -41,10 +41,6 @@ async def add_data_points(
         - Optionally updates the edge index via `index_graph_edges`.
     """
 
-
-async def add_data_points(
-    data_points: List[DataPoint], update_edge_collection: bool = True
-) -> List[DataPoint]:
     if not isinstance(data_points, list):
         raise InvalidDataPointsInAddDataPointsError("data_points must be a list.")
     if not all(isinstance(dp, DataPoint) for dp in data_points):