small button fix v2

small works but now when it's too small doesn't stop