Fixed button's bug