Entity fix

Entity fix