diff --git a/float_models/iree_exec/mnist.c b/float_models/iree_exec/mnist.c
index 7e7d519..76e8738 100644
--- a/float_models/iree_exec/mnist.c
+++ b/float_models/iree_exec/mnist.c
@@ -49,7 +49,7 @@
 }
 
 iree_hal_executable_library_query_fn_t library_query(void) {
-  return &mnist_linked_llvm_cpu_library_query;
+  return &mnist_linked_library_query;
 }
 
 iree_status_t load_input_data(const MlModel *model, void **buffer,
diff --git a/float_models/iree_exec/mobilenet_v1.c b/float_models/iree_exec/mobilenet_v1.c
index e0e8e2d..f75fe1f 100644
--- a/float_models/iree_exec/mobilenet_v1.c
+++ b/float_models/iree_exec/mobilenet_v1.c
@@ -49,7 +49,7 @@
 }
 
 iree_hal_executable_library_query_fn_t library_query(void) {
-  return &llvm_module_linked_llvm_cpu_library_query;
+  return &module_linked_library_query;
 }
 
 iree_status_t load_input_data(const MlModel *model, void **buffer,
diff --git a/quant_models/iree_exec/hps.c b/quant_models/iree_exec/hps.c
index 7353110..56d8fb5 100644
--- a/quant_models/iree_exec/hps.c
+++ b/quant_models/iree_exec/hps.c
@@ -53,7 +53,7 @@
 }
 
 iree_hal_executable_library_query_fn_t library_query(void) {
-  return &llvm_module_linked_llvm_cpu_library_query;
+  return &module_linked_library_query;
 }
 
 iree_status_t load_input_data(const MlModel *model, void **buffer,
diff --git a/quant_models/iree_exec/mobilenet_v1.c b/quant_models/iree_exec/mobilenet_v1.c
index 4c4f3be..543ef7e 100644
--- a/quant_models/iree_exec/mobilenet_v1.c
+++ b/quant_models/iree_exec/mobilenet_v1.c
@@ -49,7 +49,7 @@
 }
 
 iree_hal_executable_library_query_fn_t library_query(void) {
-  return &llvm_module_linked_llvm_cpu_library_query;
+  return &module_linked_library_query;
 }
 
 iree_status_t load_input_data(const MlModel *model, void **buffer,
diff --git a/quant_models/iree_exec/mobilenet_v2.c b/quant_models/iree_exec/mobilenet_v2.c
index 4061f54..2a27c5b 100644
--- a/quant_models/iree_exec/mobilenet_v2.c
+++ b/quant_models/iree_exec/mobilenet_v2.c
@@ -49,7 +49,7 @@
 }
 
 iree_hal_executable_library_query_fn_t library_query(void) {
-  return &llvm_module_linked_llvm_cpu_library_query;
+  return &module_linked_library_query;
 }
 
 iree_status_t load_input_data(const MlModel *model, void **buffer,
diff --git a/quant_models/iree_exec/person_detection.c b/quant_models/iree_exec/person_detection.c
index ebbd98c..911d850 100644
--- a/quant_models/iree_exec/person_detection.c
+++ b/quant_models/iree_exec/person_detection.c
@@ -63,7 +63,7 @@
 
 #if !defined(BUILD_VMVX)
 iree_hal_executable_library_query_fn_t library_query(void) {
-  return &llvm_module_linked_llvm_cpu_library_query;
+  return &module_linked_library_query;
 }
 #endif