diff --git a/src/backend/langflow/graph/graph/base.py b/src/backend/langflow/graph/graph/base.py
index dc5e6db5e..3fa726a37 100644
--- a/src/backend/langflow/graph/graph/base.py
+++ b/src/backend/langflow/graph/graph/base.py
@@ -261,8 +261,8 @@ class Graph:
 
         # Build graph and compute in-degrees
         for edge in self.edges:
-            graph[edge.source].append(edge.target)
-            in_degree[edge.target] += 1
+            graph[edge.source_id].append(edge.target_id)
+            in_degree[edge.target_id] += 1
 
         # Queue for vertices with no incoming edges
         queue = deque(vertex for vertex in self.vertices if in_degree[vertex] == 0)