diff --git a/include/mlir-hlo/Dialect/mhlo/transforms/register_passes.h b/include/mlir-hlo/Dialect/mhlo/transforms/register_passes.h
index 5c862d8..8f70f64 100644
--- a/include/mlir-hlo/Dialect/mhlo/transforms/register_passes.h
+++ b/include/mlir-hlo/Dialect/mhlo/transforms/register_passes.h
@@ -27,10 +27,10 @@ std::unique_ptr<FunctionPass> createTestInferShapedTypeMethodsPass();
 std::unique_ptr<Pass> createTestMaterializeBroadcastsPass();
 std::unique_ptr<Pass> createTestUnfuseBatchNormPass();
 
-inline void registerAllMhloPasses() {
 #define GEN_PASS_REGISTRATION
 #include "mlir-hlo/Dialect/mhlo/transforms/mhlo_passes.h.inc"
-}
+
+inline void registerAllMhloPasses() { registerMHLOPasses(); }
 
 }  // namespace mhlo
 
@@ -38,10 +38,10 @@ namespace lmhlo {
 
 std::unique_ptr<Pass> createTestLhloToLLVMPass();
 
-inline void registerAllLmhloPasses() {
 #define GEN_PASS_REGISTRATION
 #include "mlir-hlo/Dialect/mhlo/transforms/lmhlo_passes.h.inc"
-}
+
+inline void registerAllLmhloPasses() { registerLMHLOPasses(); }
 
 }  // namespace lmhlo
 }  // namespace mlir