1 files changed, 7 insertions, 5 deletions
diff --git a/crates/parser/src/lib.rs b/crates/parser/src/lib.rs
index 561d5c6..3667f1d 100644
--- a/crates/parser/src/lib.rs
+++ b/crates/parser/src/lib.rs
@@ -23,7 +23,7 @@ mod module;
 use alloc::vec::Vec;
 pub use error::*;
 use module::ModuleReader;
-use tinywasm_types::Function;
+use tinywasm_types::{Function, WasmFunction};
 use wasmparser::Validator;
 
 pub use tinywasm_types::TinyWasmModule;
@@ -108,10 +108,12 @@ impl TryFrom<ModuleReader> for TinyWasmModule {
             .code
             .into_iter()
             .zip(func_types)
-            .map(|(f, ty)| Function {
-                instructions: f.body,
-                locals: f.locals,
-                ty,
+            .map(|(f, ty)| {
+                Function::WasmFunction(WasmFunction {
+                    instructions: f.body,
+                    locals: f.locals,
+                    ty,
+                })
             })
             .collect::<Vec<_>>();