fix diagrams