diff --git a/cognee/modules/graph/utils/get_graph_from_model.py b/cognee/modules/graph/utils/get_graph_from_model.py
index a7ff3a312..4f8a2a8a9 100644
--- a/cognee/modules/graph/utils/get_graph_from_model.py
+++ b/cognee/modules/graph/utils/get_graph_from_model.py
@@ -152,10 +152,10 @@ async def get_graph_from_model(
     Returns:
         Tuple of (nodes, edges) extracted from the model
     """
-    if added_nodes == None:
+    if added_nodes is None:
         added_nodes = {}
 
-    if added_edges == None:
+    if added_edges is None:
         added_edges = {}
 
     if str(data_point.id) in added_nodes: