diff --git a/packages/react/src/utils/changes.ts b/packages/react/src/utils/changes.ts
index 5789a38a..fa6f6f30 100644
--- a/packages/react/src/utils/changes.ts
+++ b/packages/react/src/utils/changes.ts
@@ -219,7 +219,7 @@ export function applyEdgeChanges<EdgeType extends Edge = Edge>(
   return applyChanges(changes, edges) as EdgeType[];
 }
 
-export function createSelectionChange(id: string, selected: boolean) {
+export function createSelectionChange(id: string, selected: boolean): NodeSelectionChange | EdgeSelectionChange {
   return {
     id,
     type: 'select',