protected PictogramElement findPictogramElement(String id) { PictogramElement pictogramElement = null; if (id != null) { BaseElement be = bpmn2Editor.getModelHandler().findElement(id); List<PictogramElement> pes = Graphiti.getLinkService() .getPictogramElements(bpmn2Editor.getDiagramTypeProvider().getDiagram(), be); for (PictogramElement pe : pes) { if (pe instanceof ContainerShape) { pictogramElement = pe; } else if (pe instanceof FreeFormConnection) { pictogramElement = pe; } } } return pictogramElement; }