diff --git a/mlir/extras/runtime/passes.py b/mlir/extras/runtime/passes.py
index 632b6cb..6c0f1fb 100644
--- a/mlir/extras/runtime/passes.py
+++ b/mlir/extras/runtime/passes.py
@@ -142,7 +142,7 @@ def bufferize(self):
         return (
             self.Func(Pipeline().empty_tensor_to_alloc_tensor())
             .one_shot_bufferize()
-            .Func(Pipeline().finalizing_bufferize().buffer_deallocation())
+            .Func(Pipeline().buffer_deallocation())
         )
 
     def lower_to_llvm(self):