diff --git a/packages/editor-ui/src/views/canvasHelpers.ts b/packages/editor-ui/src/views/canvasHelpers.ts
index fe43ee84de..85086b8054 100644
--- a/packages/editor-ui/src/views/canvasHelpers.ts
+++ b/packages/editor-ui/src/views/canvasHelpers.ts
@@ -546,6 +546,7 @@ export const showDropConnectionState = (connection: Connection, targetPos: XYPos
 		connection.connector.setTargetPos(targetPos);
 		connection.setPaintStyle(CONNECTOR_PAINT_STYLE_PRIMARY);
 		hideOverlay(connection, OVERLAY_DROP_NODE_ID);
+		showOverlay(connection, OVERLAY_ENDPOINT_ARROW_ID);
 	}
 };
 
@@ -554,6 +555,7 @@ export const showPullConnectionState = (connection: Connection) => {
 		connection.connector.resetTargetPos();
 		connection.setPaintStyle(CONNECTOR_PAINT_STYLE_DEFAULT);
 		showOverlay(connection, OVERLAY_DROP_NODE_ID);
+		hideOverlay(connection, OVERLAY_ENDPOINT_ARROW_ID);
 	}
 };
 
@@ -561,6 +563,7 @@ export const resetConnectionAfterPull = (connection: Connection) => {
 	if (connection && connection.connector) {
 		connection.connector.resetTargetPos();
 		connection.setPaintStyle(CONNECTOR_PAINT_STYLE_DEFAULT);
+		showOverlay(connection, OVERLAY_ENDPOINT_ARROW_ID);
 	}
 };