Fix squares colors on diagrams