Fixed bug with floating button