diff --git a/src/Generator/Driver.cs b/src/Generator/Driver.cs
index 0ce29417..ff9c90c2 100644
--- a/src/Generator/Driver.cs
+++ b/src/Generator/Driver.cs
@@ -90,7 +90,7 @@ namespace Cxxi
             passes.CheckFlagEnums();
 
             if (Transform != null)
-                Transform.SetupPasses(passes);
+                Transform.SetupPasses(this, passes);
 
             passes.CleanInvalidDeclNames();
 
diff --git a/src/Generator/Library.cs b/src/Generator/Library.cs
index 3d98c24a..49468c3c 100644
--- a/src/Generator/Library.cs
+++ b/src/Generator/Library.cs
@@ -35,8 +35,9 @@ namespace Cxxi
         /// <summary>
         /// Setup your passes here.
         /// </summary>
+        /// <param name="driver"></param>
         /// <param name="passes"></param>
-        void SetupPasses(PassBuilder passes);
+        void SetupPasses(Driver driver, PassBuilder passes);
 
         /// <summary>
         /// Called to generate text at the start of the text template.