fixed error with the button