From 42cd81fd441f3b14e50431aadfc44038b8cb3464 Mon Sep 17 00:00:00 2001
From: Tal Ben-Nun <tbennun@gmail.com>
Date: Tue, 11 Oct 2022 10:52:49 +0200
Subject: [PATCH] Minor fix

---
 dace/sdfg/analysis/cfg.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/dace/sdfg/analysis/cfg.py b/dace/sdfg/analysis/cfg.py
index 2c782e9800..f2926e7bc3 100644
--- a/dace/sdfg/analysis/cfg.py
+++ b/dace/sdfg/analysis/cfg.py
@@ -225,6 +225,7 @@ def _stateorder_topological_sort(sdfg: SDFG,
         if node in visited or node is stop:
             continue
         yield node
+        visited.add(node)
 
         oe = sdfg.out_edges(node)
         if len(oe) == 0:  # End state