diff --git a/lib/Transforms/Passes.cpp b/lib/Transforms/Passes.cpp
index cb0ec4b..c9993d7 100644
--- a/lib/Transforms/Passes.cpp
+++ b/lib/Transforms/Passes.cpp
@@ -45,6 +45,7 @@ void scalehls::registerScaleHLSDSEPipeline() {
         // Legalize the input program.
         pm.addPass(scalehls::createFuncPreprocessPass(opts.hlsTopFunc));
         pm.addPass(scalehls::createMaterializeReductionPass());
+        pm.addPass(bufferization::createBufferLoopHoistingPass());
 
         // Apply the automatic design space exploration to the top function.
         pm.addPass(scalehls::createDesignSpaceExplorePass(opts.dseTargetSpec));