@Override
public void onLeave(DragContext context) {
if (context.draggable instanceof EndPoint) {
if (getDropTarget() instanceof Shape) {
Shape dropTarget = (Shape) getDropTarget();
dropTarget.hideConnectionPoints(dropTarget.diagram);
}
}
super.onLeave(context);
}