diff --git a/samples/util/util.c b/samples/util/util.c
index 731f34e..82c02c5 100644
--- a/samples/util/util.c
+++ b/samples/util/util.c
@@ -30,15 +30,17 @@
   // The buffers can be mapped on the CPU and that can also be used
   // on the device. Not all devices support this, but the ones we have now do.
 
-  iree_hal_memory_type_t input_memory_type =
-      IREE_HAL_MEMORY_TYPE_HOST_LOCAL | IREE_HAL_MEMORY_TYPE_DEVICE_VISIBLE;
+  iree_hal_buffer_params_t buffer_params = {
+      .type =
+          IREE_HAL_MEMORY_TYPE_HOST_LOCAL | IREE_HAL_MEMORY_TYPE_DEVICE_VISIBLE,
+      .access = IREE_HAL_MEMORY_ACCESS_READ,
+      .usage = IREE_HAL_BUFFER_USAGE_ALL};
   for (int i = 0; i < model->num_input; ++i) {
     if (iree_status_is_ok(result)) {
       result = iree_hal_buffer_view_wrap_or_clone_heap_buffer(
           iree_hal_device_allocator(device), model->input_shape[i],
           model->num_input_dim[i], model->hal_element_type,
-          IREE_HAL_ENCODING_TYPE_DENSE_ROW_MAJOR, input_memory_type,
-          IREE_HAL_MEMORY_ACCESS_READ, IREE_HAL_BUFFER_USAGE_ALL, *byte_span[i],
+          IREE_HAL_ENCODING_TYPE_DENSE_ROW_MAJOR, buffer_params, *byte_span[i],
           iree_allocator_null(), &(arg_buffer_views[i]));
     }
     if (byte_span[i] != NULL) {