button fix

KISS