Fix for update button