diff --git a/tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc b/tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc
index 41cce577f20..eba71da2382 100644
--- a/tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc
+++ b/tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc
@@ -69,8 +69,8 @@ inline LogicalResult GetIntegerArraySafe(
 LogicalResult ConvertMaxUnpoolingFunc::RewriteFunc() {
   func_.eraseBody();
   func_.addEntryBlock();
-  func_.setAttr(kTFImplements,
-                StringAttr::get(kMaxUnpooling, func_.getContext()));
+  func_->setAttr(kTFImplements,
+                 StringAttr::get(kMaxUnpooling, func_.getContext()));
 
   OpBuilder builder(func_.getBody());
   std::string custom_option_buffer;