fix the button problem