small button fix