Fix the buttons on query keys