Fix the buttons