Fix Class Diagram