Diagram fix