fix a miatake in last commit

fix a miatake in last commit that the button wont disappear after being
clicked