fix viewModel tests