need help to fix the button