fix color bug