updated viewmodel test