Fixed a bug with the button.