fix the color bug