fixed ViewModel bugs