Fixed the buttons