diff --git a/cognee/tests/test_delete_two_users_same_dataset.py b/cognee/tests/test_delete_two_users_same_dataset.py
index 143042eeb..c97d6eb5a 100644
--- a/cognee/tests/test_delete_two_users_same_dataset.py
+++ b/cognee/tests/test_delete_two_users_same_dataset.py
@@ -318,7 +318,7 @@ async def create_mocked_data_points(owner, for_user):
         ]  # type: ignore
     )
 
-    await record_data_in_legacy_ledger(graph_nodes, graph_edges, owner)
+    await record_data_in_legacy_ledger(graph_nodes, graph_edges)
 
     db_engine = get_relational_engine()
 
diff --git a/cognee/tests/test_delete_two_users_with_legacy_data.py b/cognee/tests/test_delete_two_users_with_legacy_data.py
index 481d5be4f..80eb7cf56 100644
--- a/cognee/tests/test_delete_two_users_with_legacy_data.py
+++ b/cognee/tests/test_delete_two_users_with_legacy_data.py
@@ -326,7 +326,7 @@ async def create_mocked_data_points(user):
         ]  # type: ignore
     )
 
-    await record_data_in_legacy_ledger(graph_nodes, graph_edges, user)
+    await record_data_in_legacy_ledger(graph_nodes, graph_edges)
 
     db_engine = get_relational_engine()