Fixed the model changes

Signed-off-by: Anurag <anurag.pattnaik@andolasoft.com>