fix the entity UT