diff --git a/packages/react/src/types/instance.ts b/packages/react/src/types/instance.ts
index 8ce74d1c..71602f92 100644
--- a/packages/react/src/types/instance.ts
+++ b/packages/react/src/types/instance.ts
@@ -140,7 +140,7 @@ export type GeneralHelpers<NodeType extends Node = Node, EdgeType extends Edge =
    */
   updateNodeData: (
     id: string,
-    dataUpdate: object | ((node: NodeType) => object),
+    dataUpdate: NodeType['data'] | ((node: NodeType) => NodeType['data']),
     options?: { replace: boolean }
   ) => void;
 };