Prepare to update viewModel if we click a button