Fixed the button for real