Fixed the entity transformer UT