Can you also change this back to `getEntityManager` like you did with the above code block?