fix EntityManager tests
Signed-off-by: Hui Zheng <fengyimzy@gmail.com>