diff --git a/src/useMergeState.ts b/src/useMergeState.ts
index 0224a28..d7280fc 100644
--- a/src/useMergeState.ts
+++ b/src/useMergeState.ts
@@ -25,7 +25,7 @@ export type MergeStateSetter<TState> = (
  * @param initialState The initial state object
  */
 export default function useMergeState<TState extends {}>(
-  initialState: TState,
+  initialState: TState | (() =>  TState),
 ): [TState, MergeStateSetter<TState>] {
   const [state, setState] = useState<TState>(initialState)