Put back the update button