diff --git a/dace/transformation/__init__.py b/dace/transformation/__init__.py
index 3a4c65efa3..0b27542ca6 100644
--- a/dace/transformation/__init__.py
+++ b/dace/transformation/__init__.py
@@ -1,3 +1,4 @@
-from .transformation import (PatternTransformation, SingleStateTransformation, MultiStateTransformation,
-                             SubgraphTransformation, ExpandTransformation, experimental_cfg_block_compatible)
+from .transformation import (PatternNode, PatternTransformation, SingleStateTransformation,
+                             MultiStateTransformation, SubgraphTransformation, ExpandTransformation,
+                             experimental_cfg_block_compatible, single_level_sdfg_only)
 from .pass_pipeline import Pass, Pipeline, FixedPointPipeline