Patched the button handler.