Not sure if that would work in your case, but you can always fall back on dictating an explicit behaviour:
Above, the icon is changing during onFocusLost / Received in the customBorder.
This also gives you direct control over where the focus goes:
Never fully understood how Escape
and Wrap
rules are supposed to work outside of containers.