Avoid undefined index notices in UOW

When auditing an entity that has a relationship that is not yet in the identity map, the UoW will cause an Undefined index error.
Fix additional edge case