diff --git a/src/test/java/org/extism/sdk/PluginTests.java b/src/test/java/org/extism/sdk/PluginTests.java
index 6ace3fe..012f0ac 100644
--- a/src/test/java/org/extism/sdk/PluginTests.java
+++ b/src/test/java/org/extism/sdk/PluginTests.java
@@ -33,14 +33,14 @@ public void shouldInvokeFunctionWithConfig() {
         var config = Map.of("key1", "value1");
         var manifest = new Manifest(List.of(CODE.pathWasmSource()), null, config);
         var output = Extism.invokeFunction(manifest, "count_vowels", "Hello World");
-        assertThat(output).isEqualTo("{\"count\": 3}");
+        assertThat(output).isEqualTo("{\"count\":3,\"total\":3,\"vowels\":\"aeiouAEIOU\"}");
     }
 
     @Test
     public void shouldInvokeFunctionFromFileWasmSource() {
         var manifest = new Manifest(CODE.pathWasmSource());
         var output = Extism.invokeFunction(manifest, "count_vowels", "Hello World");
-        assertThat(output).isEqualTo("{\"count\": 3}");
+        assertThat(output).isEqualTo("{\"count\":3,\"total\":3,\"vowels\":\"aeiouAEIOU\"}");
     }
 
     @Test
@@ -109,7 +109,7 @@ public void shouldInvokeFunctionFromUrlWasmSource() {
      public void shouldInvokeFunctionFromByteArrayWasmSource() {
          var manifest = new Manifest(CODE.byteArrayWasmSource());
          var output = Extism.invokeFunction(manifest, "count_vowels", "Hello World");
-         assertThat(output).isEqualTo("{\"count\": 3}");
+         assertThat(output).isEqualTo("{\"count\":3,\"total\":3,\"vowels\":\"aeiouAEIOU\"}");
      }
 
     @Test
@@ -131,7 +131,7 @@ public void shouldAllowInvokeFunctionFromFileWasmSourceApiUsageExample() {
 
         try (var plugin = new Plugin(manifest, false, null)) {
             var output = plugin.call(functionName, input);
-            assertThat(output).isEqualTo("{\"count\": 3}");
+            assertThat(output).contains("\"count\":3");
         }
     }
 
@@ -143,10 +143,10 @@ public void shouldAllowInvokeFunctionFromFileWasmSourceMultipleTimes() {
 
         try (var plugin = new Plugin(manifest, false, null)) {
             var output = plugin.call(functionName, input);
-            assertThat(output).isEqualTo("{\"count\": 3}");
+            assertThat(output).contains("\"count\":3");
 
             output = plugin.call(functionName, input);
-            assertThat(output).isEqualTo("{\"count\": 3}");
+            assertThat(output).contains("\"count\":3");
         }
     }
 
@@ -222,7 +222,7 @@ public void shouldAllowInvokeHostFunctionWithoutUserData() {
                 helloWorldFunction,
                 Optional.empty()
         )
-                .withNamespace("env");
+                .withNamespace("extism:host/user");
 
         HostFunction g = new HostFunction<>(
                 "hello_world",
@@ -254,7 +254,7 @@ public void shouldFailToInvokeUnknownHostFunction() {
             var plugin = new Plugin(manifest, true, null);
             plugin.call(functionName, "this is a test");
         }  catch (ExtismException e) {
-            assertThat(e.getMessage()).contains("unknown import: `env::hello_world` has not been defined");
+            assertThat(e.getMessage()).contains("unknown import: `extism:host/user::hello_world` has not been defined");
         }
     }
 
diff --git a/src/test/resources/code-functions.wasm b/src/test/resources/code-functions.wasm
index 8301227..d13506d 100644
Binary files a/src/test/resources/code-functions.wasm and b/src/test/resources/code-functions.wasm differ
diff --git a/src/test/resources/code.wasm b/src/test/resources/code.wasm
index c5fe777..5bceea0 100755
Binary files a/src/test/resources/code.wasm and b/src/test/resources/code.wasm differ