Fix examples with new IdentityMap API