diff --git a/quant_models/CMakeLists.txt b/quant_models/CMakeLists.txt
index e8dec30..b6fef98 100644
--- a/quant_models/CMakeLists.txt
+++ b/quant_models/CMakeLists.txt
@@ -52,6 +52,9 @@
     "person_detection.tflite"
   C_IDENTIFIER
     "quant_models_person_detection"
+  TFLITE_OUTPUT
+    # TODO(b/287490866): remove this argument after IREE fix is landed
+    "--output-array=MobilenetV1/Logits/SpatialSqueeze"
   FLAGS
     "-iree-input-type=tosa"
     "-riscv-v-fixed-length-vector-lmul-max=8"
diff --git a/quant_models/iree_exec/person_detection_test.filecheck b/quant_models/iree_exec/person_detection_test.filecheck
index 1a880de..557f3bf 100644
--- a/quant_models/iree_exec/person_detection_test.filecheck
+++ b/quant_models/iree_exec/person_detection_test.filecheck
@@ -1 +1,2 @@
-// CHECK: {{Non-person Score: -11[3-4]; Person Score: 11[3-4]}}
+// TODO(b/287490866): change range after IREE fix is landed
+// CHECK: {{Non-person Score: -11[2-4]; Person Score: 11[0-4]}}