fix viewmodel