As far as I know it doesn’t work with the core entity button yet (or I don’t know how to do it, as it isn’t documented yet).
I use the custom:button-card for this (which is much more customizable and flexible). I’d suggest you give it a try.
Button Card: https://github.com/custom-cards/button-card
iOS HA beta: Home Assistant for iOS 2.0 Beta Testing