fix the buttons