diff --git a/cognee/modules/graph/methods/has_data_related_nodes.py b/cognee/modules/graph/methods/has_data_related_nodes.py
index f3c1e29b4..0cdb6173c 100644
--- a/cognee/modules/graph/methods/has_data_related_nodes.py
+++ b/cognee/modules/graph/methods/has_data_related_nodes.py
@@ -13,4 +13,4 @@ async def has_data_related_nodes(dataset_id: UUID, data_id: UUID, session: Async
     )
 
     data_related_node = await session.scalar(query_statement)
-    return data_related_node != None
+    return data_related_node is not None
diff --git a/cognee/tests/test_delete_default_graph_with_legacy_data_2.py b/cognee/tests/test_delete_default_graph_with_legacy_data_2.py
index f9ec9434f..082d250f5 100644
--- a/cognee/tests/test_delete_default_graph_with_legacy_data_2.py
+++ b/cognee/tests/test_delete_default_graph_with_legacy_data_2.py
@@ -288,10 +288,7 @@ async def main(mock_create_structured_output: AsyncMock):
     )
     johns_data_id = add_john_result.data_ingestion_info[0]["data_id"]
 
-    add_marie_result = await cognee.add(
-        "Marie works for Apple as well. She is a software engineer on MacOS project."
-    )
-    # maries_data_id = add_marie_result.data_ingestion_info[0]["data_id"]
+    await cognee.add("Marie works for Apple as well. She is a software engineer on MacOS project.")
 
     cognify_result: dict = await cognee.cognify()
     dataset_id = list(cognify_result.keys())[0]