Fixed the buttons!