Add the missing IViewModel