diff --git a/src/ColorFunctionContext.tsx b/src/ColorFunctionContext.tsx
index d76e966..06f1798 100644
--- a/src/ColorFunctionContext.tsx
+++ b/src/ColorFunctionContext.tsx
@@ -23,11 +23,12 @@ export function ColorFunctionProvider({ children }: WasmProviderProps) {
   const [error, setError] = useState<Error | null>(null);
 
   const contextValue = useMemo(
-    () => ({
-      colorFn: wasmInstance,
-      isLoading: isPending,
-      error,
-    }),
+    () =>
+      ({
+        colorFn: wasmInstance,
+        isLoading: isPending,
+        error,
+      } as ColorFunctionContextType),
     [wasmInstance, isPending, error],
   );
   useEffect(() => {