fix the button in main