diff --git a/lib/TPP/Transforms/HoistVectorTransfers.cpp b/lib/TPP/Transforms/HoistVectorTransfers.cpp
index 118a826fb..0faf49a40 100644
--- a/lib/TPP/Transforms/HoistVectorTransfers.cpp
+++ b/lib/TPP/Transforms/HoistVectorTransfers.cpp
@@ -226,10 +226,10 @@ struct HoistVectorTransferOp : OpRewritePattern<vector::ContractionOp> {
     // Erase the old vector contract operation
     for (auto result : contractOp->getResults()) {
       for (auto *userOp : result.getUsers()) {
-        userOp->erase();
+        rewriter.eraseOp(userOp);
       }
     }
-    contractOp.erase();
+    rewriter.eraseOp(contractOp);
 
     return success();
   }