diff --git a/index.d.ts b/index.d.ts
index db01dc2..e04879c 100644
--- a/index.d.ts
+++ b/index.d.ts
@@ -207,7 +207,7 @@ declare module "@uidotdev/usehooks" {
 
   export function useNetworkState(): NetworkState;
 
-  export function useObjectState<T>(initialValue: T): [T, (arg: T) => void];
+  export function useObjectState<T>(initialValue: T): [T, (arg: T | ((arg: (arg: T) => Partial<T>) => void)) => void];
 
   export function useOrientation(): {
     angle: number;