Fix UML-diagram