Fixes bug with viewModelBase.