diff --git a/CMakeLists.txt b/CMakeLists.txt
index b54cb7e..f932680 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -41,6 +41,7 @@
 set(IREE_BUILD_BINDINGS_TFLITE OFF)
 set(IREE_HAL_DRIVER_LOCAL_SYNC ON CACHE BOOL "" FORCE)
 set(IREE_HAL_EXECUTABLE_LOADER_DEFAULTS OFF CACHE BOOL "" FORCE)
+set(IREE_HAL_EXECUTABLE_LOADER_VMVX_MODULE ON CACHE BOOL "" FORCE)
 set(IREE_HAL_DRIVER_DEFAULTS OFF CACHE BOOL "" FORCE)
 
 #-------------------------------------------------------------------------------
diff --git a/samples/util/util.c b/samples/util/util.c
index 858a0ec..a031dba 100644
--- a/samples/util/util.c
+++ b/samples/util/util.c
@@ -67,7 +67,7 @@
 }
 
 iree_status_t run(const MlModel *model) {
-  IREE_RETURN_IF_ERROR(iree_hal_module_register_types());
+  IREE_RETURN_IF_ERROR(iree_hal_module_register_all_types());
 
   iree_vm_instance_t *instance = NULL;
   iree_status_t result =