viewmodel fix