Fixing "Not an entity"3