Fixed class diagram