Shouldn't this use the viewmodel?