Model update

bug fix