diff --git a/dace/transformation/passes/prune_symbols.py b/dace/transformation/passes/prune_symbols.py
index a01d903a1d..c501a769ff 100644
--- a/dace/transformation/passes/prune_symbols.py
+++ b/dace/transformation/passes/prune_symbols.py
@@ -111,6 +111,8 @@ def used_symbols(self, sdfg: SDFG) -> Set[str]:
                         if node.code_exit.language != dtypes.Language.Python:
                             result |= symbolic.symbols_in_code(node.code_exit.as_string, sdfg.symbols.keys(),
                                                             node.ignored_symbols)
+            else:
+                result |= block.used_symbols(all_symbols=True, with_contents=False)
 
         for e in sdfg.all_interstate_edges():
             result |= e.data.free_symbols