src/vm.ts

@@ -26,7 +26,9 @@ export class VM {
     this.scope = new Scope()
   }
 
-  registerFunction(name: string, fn: Function) {
+  registerFunction(name: string, fn: NativeFunction | Function) {
+    // If it's already a NativeFunction, use it directly
+    // Otherwise, assume it's a JS function and wrap it
     const wrapped = isWrapped(fn) ? fn as NativeFunction : wrapNative(fn)
     this.nativeFunctions.set(name, wrapped)
   }

tests/native.test.ts

@@ -128,7 +128,7 @@ test("CALL_NATIVE - function not found", async () => {
 
   const vm = new VM(bytecode)
 
-  expect(vm.run()).rejects.toThrow('CALL_NATIVE: function not found: nonexistent')
+  await expect(vm.run()).rejects.toThrow('CALL_NATIVE: function not found: nonexistent')
 })
 
 test("CALL_NATIVE - using result in subsequent operations", async () => {