diff --git a/src/finn/analysis/topology.py b/src/finn/analysis/topology.py
index a94342565012804159774c6c55aa580bf1b249ba..3973c8d26d417e009463fc22e041e5999ae2c371 100644
--- a/src/finn/analysis/topology.py
+++ b/src/finn/analysis/topology.py
@@ -91,7 +91,7 @@ def nodes_in_expected_order(model):
 
     all_OK = True
     for n in model.graph.node:
-        successors = model.find_successors(n)
+        successors = model.find_direct_successors(n)
         if successors is not None:
             for successor in successors:
                 # check the condition by checking the antithesis