fix the button bug