diff --git a/model_util/util.c b/model_util/util.c
index df0056d..a390e84 100644
--- a/model_util/util.c
+++ b/model_util/util.c
@@ -89,9 +89,9 @@
   // Create hal_module
   iree_vm_module_t *hal_module = NULL;
   if (iree_status_is_ok(result)) {
-    result =
-        iree_hal_module_create(instance, *device, IREE_HAL_MODULE_FLAG_NONE,
-                               host_allocator, &hal_module);
+    result = iree_hal_module_create(instance, /*device_count=*/1, device,
+                                    IREE_HAL_MODULE_FLAG_NONE, host_allocator,
+                                    &hal_module);
   }
   iree_vm_module_t *modules[] = {hal_module, module};
 #endif