diff --git a/src/types/edge.ts b/src/types/edge.ts
index f5912449..0fd595cd 100644
--- a/src/types/edge.ts
+++ b/src/types/edge.ts
@@ -135,6 +135,8 @@ export interface EdgeProps<Data = any> {
 
 export interface SmoothStepEdgeProps<Data = any> extends EdgeProps<Data> {
   id: string
+  sourceNode: any
+  targetNode: any
   label?:
     | string
     | {