diff --git a/dace/sdfg/graph.py b/dace/sdfg/graph.py
index 5c93149529..91ed698896 100644
--- a/dace/sdfg/graph.py
+++ b/dace/sdfg/graph.py
@@ -526,10 +526,10 @@ def edges(self):
         return [DiGraph._from_nx(e) for e in self._nx.edges()]
 
     def in_edges(self, node):
-        return [DiGraph._from_nx(e) for e in self._nx.in_edges()]
+        return [DiGraph._from_nx(e) for e in self._nx.in_edges(node, True)]
 
     def out_edges(self, node):
-        return [DiGraph._from_nx(e) for e in self._nx.out_edges()]
+        return [DiGraph._from_nx(e) for e in self._nx.out_edges(node, True)]
 
     def add_node(self, node):
         return self._nx.add_node(node)