fix update button issue