Fix button class