Fix to update button