Diagram correction