I'm trying to catch any error cases in the ViewModel and update the UIState, but it's not wor