diff --git a/mlir/test/Conversion/TosaToArith/tosa-to-arith.mlir b/mlir/test/Conversion/TosaToArith/tosa-to-arith.mlir
--- a/mlir/test/Conversion/TosaToArith/tosa-to-arith.mlir
+++ b/mlir/test/Conversion/TosaToArith/tosa-to-arith.mlir
@@ -118,10 +118,13 @@
   %res = tosa.apply_scale %arg0, %arg1, %arg2 {double_round = true} : (i48, i32, i8) -> i32
   return %res : i32
 }
+
+// -----
+
 // CHECK-LABEL: @apply_scale_test_i64
 // SCALE: tosa.apply_scale
 func.func @apply_scale_test_i64(%arg0 : i64, %arg1 : i32, %arg2 : i8) -> (i32) {
-  // CHECK-DAG: %[[C0:.+]] = arith.constant 0 : i48
+  // CHECK-DAG: %[[C0:.+]] = arith.constant 0 : i64
   // CHECK-DAG: %[[C1:.+]] = arith.constant 1 : i64
   // CHECK-DAG: %[[C31:.+]] = arith.constant 31 : i32