diff --git a/tests/filecheck/array.filecheck.py b/tests/filecheck/array.filecheck.py
index d22faa243b9b..ba2e3392ecbe 100644
--- a/tests/filecheck/array.filecheck.py
+++ b/tests/filecheck/array.filecheck.py
@@ -77,9 +77,7 @@ def main(_):
 
   # CHECK-LABEL: TEST: select bool[2,7] int32[2,7] int32[2,7]
   # CHECK: mhlo.select
-  # CHECK-SAME: tensor<2x7xi1>
-  # CHECK-SAME: tensor<2x7xi32>
-  # CHECK-SAME: tensor<2x7xi32>
+  # CHECK-SAME: tensor<2x7xi1>, tensor<2x7xi32>
   print_ir(np.empty([2, 7], np.bool_), np.empty([2, 7], np.int32),
            np.empty([2, 7], np.int32))(lax.select)