Might want to make sure `button` is truthy too