Fix UML Diagram