Small update viewmodel fix