Fix button states