Change the logic of when to show the buttons, and test.