Entity fix III

Entity fix III