Update EntityManager unit tests