Fix update for button