fix button class