Fix polyfill

tc39 · Feb 12, 2024 · b4a83fa · b4a83fa
1 parent 18f043e
commit b4a83fa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/promise-polyfill.ts b/src/promise-polyfill.ts
@@ -3,10 +3,11 @@ import { AsyncContext } from "./index";
 import type { AnyFunc } from "./types";
 
 export const nativeThen = Promise.prototype.then;
+const { wrap } = AsyncContext.Snapshot;
 
 function wrapFn<F extends AnyFunc<any>>(fn: F | null | undefined) {
   if (typeof fn !== "function") return undefined;
-  return AsyncContext.wrap(fn);
+  return wrap(fn);
 }
 
 export function then<T>(