diff --git a/mlir/test/Target/llvmir.mlir b/mlir/test/Target/llvmir.mlir
index 89f71477325dd0..31d5ecf0a55116 100644
--- a/mlir/test/Target/llvmir.mlir
+++ b/mlir/test/Target/llvmir.mlir
@@ -1265,7 +1265,7 @@ llvm.func @cond_br_weights(%cond : !llvm.i1, %arg0 : !llvm.i32,  %arg1 : !llvm.i
   llvm.return %arg1 : !llvm.i32
 }
 
-// CHECK: ![[NODE]] = !{!"branch_weights", i32 5, i32 10}
+// CHECK: ![[NODE]] = !{!"branch_weights", i64 5, i64 10}
 
 // -----