private void updateIcon(EObject element) { String className = entry.getMessage(); IEditorPart editorPart = PlatformUI.getWorkbench().getActiveWorkbenchWindow().getActivePage().getActiveEditor(); if (editorPart instanceof IDiagramWorkbenchPart) { DiagramEditPart diagramPart = ((IDiagramWorkbenchPart) editorPart).getDiagramEditPart(); List<EditPart> editPartCollector = new ArrayList<EditPart>(); Neuro4jDiagramEditorUtil.findElementsInDiagramByID(diagramPart, element, editPartCollector); if (editPartCollector.size() > 0) { LogicNodeEditPart editpart = (LogicNodeEditPart) editPartCollector.get(0); editpart.updateImageForClass(className); } // editPartCollector } }