bug fix on update button