Remove isEmbeddedDocument checks in UnitOfWork.