diff --git a/src/index.ts b/src/index.ts index a729473..e63bdce 100644 --- a/src/index.ts +++ b/src/index.ts @@ -14,13 +14,13 @@ export function useImmer(initialValue: any) { true ) ); - return [ - val, - useCallback((updater) => { - if (typeof updater === "function") updateValue(produce(updater)); - else updateValue(freeze(updater)); - }, []), - ]; + + const setter = useCallback((updater) => { + if (typeof updater === "function") updateValue(produce(updater)); + else updateValue(freeze(updater)); + }, []), + + return [val, setter]; } // Provides different overloads of `useImmerReducer` similar to `useReducer` from `@types/react`.