minor button fix