Fix class diagram.