fixed the button problem