Fixed the bug causing the button not to work.