diff --git a/quarkus-proxy-wasm-example/src/main/java/io/roastedroot/proxywasm/jaxrs/example/App.java b/quarkus-proxy-wasm-example/src/main/java/io/roastedroot/proxywasm/jaxrs/example/App.java
index 9071458..cdfdbb2 100644
--- a/quarkus-proxy-wasm-example/src/main/java/io/roastedroot/proxywasm/jaxrs/example/App.java
+++ b/quarkus-proxy-wasm-example/src/main/java/io/roastedroot/proxywasm/jaxrs/example/App.java
@@ -1,5 +1,6 @@
 package io.roastedroot.proxywasm.jaxrs.example;
 
+import com.dylibso.chicory.experimental.aot.AotMachine;
 import com.dylibso.chicory.wasm.Parser;
 import com.dylibso.chicory.wasm.WasmModule;
 import io.roastedroot.proxywasm.StartException;
@@ -23,6 +24,7 @@ public PluginFactory example() throws StartException {
                 Plugin.builder()
                         .withName("example")
                         .withPluginConfig("{ \"type\": \"headerTests\" }")
+                        .withMachineFactory(AotMachine::new)
                         .build(module);
     }
 }
diff --git a/quarkus-proxy-wasm/integration-tests/src/main/java/io/quarkiverse/proxywasm/it/App.java b/quarkus-proxy-wasm/integration-tests/src/main/java/io/quarkiverse/proxywasm/it/App.java
index ccf20bd..41d2db3 100644
--- a/quarkus-proxy-wasm/integration-tests/src/main/java/io/quarkiverse/proxywasm/it/App.java
+++ b/quarkus-proxy-wasm/integration-tests/src/main/java/io/quarkiverse/proxywasm/it/App.java
@@ -1,5 +1,6 @@
 package io.quarkiverse.proxywasm.it;
 
+import com.dylibso.chicory.experimental.aot.AotMachine;
 import com.dylibso.chicory.wasm.Parser;
 import com.dylibso.chicory.wasm.WasmModule;
 import com.google.gson.Gson;
@@ -39,6 +40,7 @@ public PluginFactory headerTestsNotShared() throws StartException {
                         .withShared(false)
                         .withLogger(new MockLogger("headerTestsNotShared"))
                         .withPluginConfig(gson.toJson(Map.of("type", "headerTests")))
+                        .withMachineFactory(AotMachine::new)
                         .build(parseTestModule("/go-examples/unit_tester/main.wasm"));
     }
 
@@ -49,6 +51,7 @@ public PluginFactory tickTests() throws StartException {
                         .withName("tickTests")
                         .withLogger(new MockLogger("tickTests"))
                         .withPluginConfig(gson.toJson(Map.of("type", "tickTests")))
+                        .withMachineFactory(AotMachine::new)
                         .build(parseTestModule("/go-examples/unit_tester/main.wasm"));
     }
 
@@ -60,6 +63,7 @@ public PluginFactory ffiTests() throws StartException {
                         .withLogger(new MockLogger("ffiTests"))
                         .withPluginConfig(gson.toJson(Map.of("type", "ffiTests")))
                         .withForeignFunctions(Map.of("reverse", App::reverse))
+                        .withMachineFactory(AotMachine::new)
                         .build(parseTestModule("/go-examples/unit_tester/main.wasm"));
     }
 
@@ -84,6 +88,7 @@ public PluginFactory httpCallTests() throws StartException {
                                                 "upstream", "web_service",
                                                 "path", "/ok")))
                         .withUpstreams(Map.of("web_service", "localhost:8081"))
+                        .withMachineFactory(AotMachine::new)
                         .build(parseTestModule("/go-examples/unit_tester/main.wasm"));
     }
 }