Fix breaking the button.. :|