Fix color update bug