diff --git a/src/bloqade/stim/passes/flatten.py b/src/bloqade/stim/passes/flatten.py
index 3d86192a..87002a95 100644
--- a/src/bloqade/stim/passes/flatten.py
+++ b/src/bloqade/stim/passes/flatten.py
@@ -5,7 +5,7 @@
 from kirin.passes import Pass
 from kirin.rewrite.abc import RewriteResult
 
-from bloqade.qasm2.passes.fold import AggressiveUnroll
+from bloqade.rewrite.passes import AggressiveUnroll
 from bloqade.stim.passes.simplify_ifs import StimSimplifyIfs