Fix fab button behavior