Fix the button