fix StateButton glitches