diff --git a/src/hooks/useNodesEdgesState.ts b/src/hooks/useNodesEdgesState.ts
index 13971d76..ad857568 100644
--- a/src/hooks/useNodesEdgesState.ts
+++ b/src/hooks/useNodesEdgesState.ts
@@ -12,16 +12,16 @@ function createUseItemsState(
   applyChanges: ApplyChanges<Node, NodeChange>
 ): <NodeData = any>(
   initialItems: Node<NodeData>[]
-) => [Node<NodeData>[], Dispatch<SetStateAction<Node<NodeData>>>, OnChange<NodeChange>];
+) => [Node<NodeData>[], Dispatch<SetStateAction<Node<NodeData>[]>>, OnChange<NodeChange>];
 function createUseItemsState(
   applyChanges: ApplyChanges<Edge, EdgeChange>
 ): <EdgeData = any>(
   initialItems: Edge<EdgeData>[]
-) => [Edge<EdgeData>[], Dispatch<SetStateAction<Edge<EdgeData>>>, OnChange<EdgeChange>];
+) => [Edge<EdgeData>[], Dispatch<SetStateAction<Edge<EdgeData>[]>>, OnChange<EdgeChange>];
 function createUseItemsState(
   applyChanges: ApplyChanges<any, any>
-): (initialItems: any[]) => [any[], Dispatch<SetStateAction<any>>, OnChange<any>] {
-  return (initialItems: any) => {
+): (initialItems: any[]) => [any[], Dispatch<SetStateAction<any[]>>, OnChange<any>] {
+  return (initialItems: any[]) => {
     const [items, setItems] = useState(initialItems);
 
     const onItemsChange = useCallback((changes: any[]) => setItems((items: any) => applyChanges(changes, items)), []);