Fix button not reappearing