patch(context): use internally

ealush · Jun 14, 2022 · b11c779 · b11c779 · vercel · Jun 14, 2022
1 parent 44721e6
commit b11c779
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/packages/context/src/context.ts b/packages/context/src/context.ts
@@ -22,11 +22,12 @@ export function createContext<T extends Record<string, unknown>>(
   };
 
   function useX(errorMessage?: string): T {
+    const ctx = use();
     invariant(
-      storage.ctx,
+      ctx,
       defaultTo(errorMessage, 'Context was used after it was closed')
     );
-    return storage.ctx as T;
+    return ctx;
   }
 
   function run<R>(ctxRef: Partial<T>, fn: (context: T) => R): R {