Entity fix II

Entity fix II