Button fix: don't query all buttons in main.