Hi @sm-rob
I think its worth noticing which context you are consuming. In the issue you consumed a Workspace Context. The Entity Context is a different one, use this Context Token for it.
UMB_PARENT_ENTITY_CONTEXT
Hopefully that is the problem you experience?