Fixing "Not an entity"4