diff --git a/src/components/Handle/index.tsx b/src/components/Handle/index.tsx
index f7c2b9ca..459380aa 100644
--- a/src/components/Handle/index.tsx
+++ b/src/components/Handle/index.tsx
@@ -16,7 +16,8 @@ interface HandleProps {
   type: HandleType;
   position: Position;
   onConnect?: OnConnectFunc;
-  isValidConnection?: () => boolean;
+  isValidConnection?: (connection: Connection) => boolean;
+  id?: string;
 }
 
 const Handle = memo(