tiny update to include view model