diff --git a/model_util/util.c b/model_util/util.c
index 4833243..187a66a 100644
--- a/model_util/util.c
+++ b/model_util/util.c
@@ -226,7 +226,7 @@
       result = iree_hal_buffer_map_range(
           iree_hal_buffer_view_buffer(ret_buffer_view),
           IREE_HAL_MAPPING_MODE_SCOPED, IREE_HAL_MEMORY_ACCESS_READ, 0,
-          IREE_WHOLE_BUFFER, &mapped_memories[index_output]);
+          IREE_HAL_WHOLE_BUFFER, &mapped_memories[index_output]);
     }
 
     if (iree_status_is_ok(result)) {