fix view model bug