Fix update button bug.