Fix minor button class issue