Fix floating button class