diff --git a/packages/xstate-graph/src/graph.ts b/packages/xstate-graph/src/graph.ts
index 26f584a98c..5c31f66af9 100644
--- a/packages/xstate-graph/src/graph.ts
+++ b/packages/xstate-graph/src/graph.ts
@@ -95,8 +95,8 @@ export function getEventEdges<
               : node;
 
             return {
-              source: node,
-              target: targetNode,
+              source: node as StateNode<TContext, any, any>, // TODO: fix
+              target: targetNode as StateNode<TContext, any, any>, // TODO: fix
               event,
               actions: transition.actions
                 ? transition.actions.map(getActionType)