Update stub to use the models.