Fix the coloring bug