Adding remove mocking to entityManager