diff --git a/float_models/CMakeLists.txt b/float_models/CMakeLists.txt
index 6c57aa1..9bf848a 100644
--- a/float_models/CMakeLists.txt
+++ b/float_models/CMakeLists.txt
@@ -29,7 +29,7 @@
   C_IDENTIFIER
     "float_models_mnist"
   FLAGS
-    "-iree-input-type=mhlo"
+    "-iree-input-type=mhlo_legacy"
   RVV_OFF
 )