diff --git a/package/src/store/actions.ts b/package/src/store/actions.ts
index 09514f8a..ec806040 100644
--- a/package/src/store/actions.ts
+++ b/package/src/store/actions.ts
@@ -249,6 +249,8 @@ export default (state: State, getters: ComputedGetters): Actions => {
           ...state.defaultEdgeOptions,
           ...storedEdge,
         }),
+        sourceNode,
+        targetNode,
       })
 
       return res
@@ -287,6 +289,8 @@ export default (state: State, getters: ComputedGetters): Actions => {
         state.edges.push({
           ...state.defaultEdgeOptions,
           ...edge,
+          sourceNode,
+          targetNode,
         })
       }
     })
diff --git a/package/src/types/edge.ts b/package/src/types/edge.ts
index 20f5a0f4..36079a7d 100644
--- a/package/src/types/edge.ts
+++ b/package/src/types/edge.ts
@@ -97,6 +97,8 @@ export interface EdgePositions {
 export type GraphEdge<Data = ElementData> = Edge<Data> & {
   selected?: boolean
   z?: number
+  sourceNode: GraphNode
+  targetNode: GraphNode
 } & EdgePositions
 
 /** these props are passed to edge components */