Tweak action button behavior