small bugfix for the button