diff --git a/ml-proto/test/has_feature.wast b/ml-proto/test/has_feature.wast
index e69be15f7c..529c34143b 100644
--- a/ml-proto/test/has_feature.wast
+++ b/ml-proto/test/has_feature.wast
@@ -1,12 +1,9 @@
 (module
-  (func $has_wasm (result i32)
-    (has_feature "wasm"))
-  (export "has_wasm" $has_wasm)
-
-  (func $has_simd128 (result i32)
-    (has_feature "simd128"))
-  (export "has_simd128" $has_simd128)
+  (import $assert_eq "assert" "eq_i32" (param i32) (param i32))
+  (fund $main
+    (call_import $assert_eq (has_feature "simd128") (i32.const 1))
+    (call_import $assert_eq (has_feature "wasm") (i32.const 0))
+  )
 )
 
-(assert_return (invoke "has_wasm") (i32.const 1))
-(assert_return (invoke "has_simd128") (i32.const 0))
+
diff --git a/ml-proto/test/imports.wast b/ml-proto/test/imports.wast
index 6e11899461..35e4f01787 100644
--- a/ml-proto/test/imports.wast
+++ b/ml-proto/test/imports.wast
@@ -17,9 +17,9 @@
             (f64.const 53)
         )
     )
-    (export "print32" $print32)
-    (export "print64" $print64)
+    (func $main
+        (call $print32 (i32.const 13))
+        (call $print64 (i64.const 24))
+    )
 )
 
-(invoke "print32" (i32.const 13))
-(invoke "print64" (i64.const 24))