Fix bug how the color is set