fixed an error with the button