minor, button class fix