Fix drawing the button