fixed bug with model update
0.2.2