Fix coloring on diaganoal