bug fix button class