diff --git a/include/Dialect/TensorExt/IR/TensorExtPatterns.td b/include/Dialect/TensorExt/IR/TensorExtPatterns.td
index b3abd1f93..8a6586788 100644
--- a/include/Dialect/TensorExt/IR/TensorExtPatterns.td
+++ b/include/Dialect/TensorExt/IR/TensorExtPatterns.td
@@ -6,19 +6,16 @@ include "mlir/Dialect/Arith/IR/ArithOps.td"
 include "mlir/Dialect/Tensor/IR/TensorOps.td"
 include "mlir/IR/PatternBase.td"
 
-def GetZeroAttr : NativeCodeCall<"$_builder.getIndexAttr(0)">;
-
 def InsertRotations : Pattern<
   (Arith_AddIOp
     (Tensor_ExtractOp $t1, (variadic $i1)),
     (Tensor_ExtractOp $t2, (variadic $i2)),
     $overflow),
   [
-    (Arith_ConstantOp:$zero (GetZeroAttr)),
     (TensorExt_RotateOp:$r1 $t1, $i1),
     (TensorExt_RotateOp:$r2 $t2, $i2),
     (Arith_AddIOp:$addResult $r1, $r2, $overflow),
-    (Tensor_ExtractOp $addResult, $zero),
+    (Tensor_ExtractOp $addResult, (Arith_ConstantOp ConstantAttr<IndexAttr, "0">)),
   ]
 >;