diff --git a/python/scalehls/transforms.py b/python/scalehls/transforms.py
index c4de1641..114e3c95 100644
--- a/python/scalehls/transforms.py
+++ b/python/scalehls/transforms.py
@@ -146,13 +146,17 @@ def apply_convert_dataflow_to_func_passes(module: Module):
     pm = PassManager.parse(
         "builtin.module("
         "func.func("
+        "scalehls-generate-directives,"
+        "convert-linalg-to-loops,"
+        "fold-memref-alias-ops,"
         "scalehls-raise-scf-to-affine,"
-        "convert-linalg-to-affine-loops,"
         "affine-loop-normalize,"
-        "affine-simplify-structures"
+        "affine-simplify-structures,"
+        "scalehls-apply-directives,"
+        "affine-scalrep"
         "),"
-        "fold-memref-alias-ops,"
         "scalehls-convert-dataflow-to-func,"
+        # "buffer-results-to-out-params,"
         "cse, canonicalize"
         ")")
     pm.run(module.operation)