Fix gradient coloring bug