Refactored one of ViewModel tests