Fixed a button_click bug