static EntityType getTypeIfExisting(ActivityDiagram system, String code) { if (system.entityExist(code)) { final IEntity ent = system.entities().get(code); if (ent.getType() == EntityType.BRANCH) { return EntityType.BRANCH; } } return EntityType.ACTIVITY; }