Fix button color