Fix the button layout