Fix color gradient bug