diff --git a/src/Diagram/Port/Port.js b/src/Diagram/Port/Port.js
index 9bba74e..b902659 100644
--- a/src/Diagram/Port/Port.js
+++ b/src/Diagram/Port/Port.js
@@ -29,7 +29,7 @@ const Port = (props) => {
   onDragEnd((event) => {
     const targetPort = event.target.getAttribute('data-port-id');
     if (targetPort && event.target !== ref.current && canLink(id, targetPort, type) && onSegmentConnect) {
-      const args = type === 'input' ? [id, targetPort, type] : [targetPort, id, type];
+      const args = [id, targetPort, type];
 
       onSegmentConnect(...args);
       return;