Fix reference to view-model