Fix minor button bug