diff --git a/packages/hooks/src/useGetState/demo/demo1.tsx b/packages/hooks/src/useGetState/demo/demo1.tsx
index 119712b36e..8a50a86909 100644
--- a/packages/hooks/src/useGetState/demo/demo1.tsx
+++ b/packages/hooks/src/useGetState/demo/demo1.tsx
@@ -22,5 +22,5 @@ export default () => {
     };
   }, []);
 
-  return <button onClick={() => setCount((count) => count + 1)}>count: {count}</button>;
+  return <button onClick={() => setCount((prevCount) => prevCount + 1)}>count: {count}</button>;
 };
diff --git a/packages/hooks/src/useGetState/index.ts b/packages/hooks/src/useGetState/index.ts
index 51f76c61aa..64a1ed56f9 100644
--- a/packages/hooks/src/useGetState/index.ts
+++ b/packages/hooks/src/useGetState/index.ts
@@ -1,20 +1,26 @@
-import type { Dispatch, SetStateAction } from 'react';
+import type { Dispatch } from 'react';
 import { useState, useRef, useCallback } from 'react';
+import { isFunction } from '../utils';
 
+type SetStateAction<S> = Dispatch<React.SetStateAction<S>>;
 type GetStateAction<S> = () => S;
 
-function useGetState<S>(
-  initialState: S | (() => S),
-): [S, Dispatch<SetStateAction<S>>, GetStateAction<S>];
+function useGetState<S>(initialState: S | (() => S)): [S, SetStateAction<S>, GetStateAction<S>];
 function useGetState<S = undefined>(): [
   S | undefined,
-  Dispatch<SetStateAction<S | undefined>>,
+  SetStateAction<S | undefined>,
   GetStateAction<S | undefined>,
 ];
 function useGetState<S>(initialState?: S) {
-  const [state, setState] = useState(initialState);
+  const [state, setInnerState] = useState(initialState);
   const stateRef = useRef(state);
-  stateRef.current = state;
+
+  const setState = useCallback<SetStateAction<S | undefined>>((stateOrAction) => {
+    const newState = isFunction(stateOrAction) ? stateOrAction(stateRef.current) : stateOrAction;
+
+    stateRef.current = newState;
+    setInnerState(newState);
+  }, []);
 
   const getState = useCallback(() => stateRef.current, []);