Trying to fix issues with buttons.