Add missing ViewModel