diff --git a/src/components/ConnectionLine/index.tsx b/src/components/ConnectionLine/index.tsx
index bb9161bb4..e3c71f9e3 100644
--- a/src/components/ConnectionLine/index.tsx
+++ b/src/components/ConnectionLine/index.tsx
@@ -79,6 +79,8 @@ export default ({
           targetPosition={targetPosition}
           connectionLineType={connectionLineType}
           connectionLineStyle={connectionLineStyle}
+          sourceNode={sourceNode}
+          sourceHandle={sourceHandle}
         />
       </g>
     );
diff --git a/src/types/index.ts b/src/types/index.ts
index 962477955..8006aa1a2 100644
--- a/src/types/index.ts
+++ b/src/types/index.ts
@@ -303,6 +303,8 @@ export type ConnectionLineComponentProps = {
   targetPosition?: Position;
   connectionLineStyle?: CSSProperties;
   connectionLineType: ConnectionLineType;
+  sourceNode?: Node;
+  sourceHandle?: HandleElement;
 };
 
 export type ConnectionLineComponent = React.ComponentType<ConnectionLineComponentProps>;