Fix button state