diff --git a/lib/Hello/LowerToAffine.cpp b/lib/Hello/LowerToAffine.cpp
index b73a061049..abc7456382 100644
--- a/lib/Hello/LowerToAffine.cpp
+++ b/lib/Hello/LowerToAffine.cpp
@@ -135,7 +135,7 @@ class PrintOpLowering : public mlir::OpConversionPattern<hello::PrintOp> {
                   mlir::ConversionPatternRewriter &rewriter) const final {
     // We don't lower "hello.print" in this pass, but we need to update its
     // operands.
-    rewriter.updateRootInPlace(op,
+    rewriter.modifyOpInPlace(op,
                                [&] { op->setOperands(adaptor.getOperands()); });
     return mlir::success();
   }