Fix button enabled bug