diff --git a/lib/Dialect/mhlo/IR/hlo_ops.cc b/lib/Dialect/mhlo/IR/hlo_ops.cc
index 706f2e9..f91c9a1 100644
--- a/lib/Dialect/mhlo/IR/hlo_ops.cc
+++ b/lib/Dialect/mhlo/IR/hlo_ops.cc
@@ -1362,7 +1362,8 @@ OpFoldResult OrOp::fold(ArrayRef<Attribute> operands) {
 OpFoldResult XorOp::fold(ArrayRef<Attribute> operands) {
   auto rType = getType().cast<ShapedType>();
   if (lhs() == rhs()) {
-    return DenseIntElementsAttr::get(rType, 0);
+    Builder builder(getContext());
+    return builder.getZeroAttr(rType);
   }
 
   auto lhsVal = operands[0].dyn_cast_or_null<DenseElementsAttr>();