fix viewmodel tests