Minor button fix