Trying to fix the button