Fix colours for the color_map