fix the button.