Fix the button color