Fix use statements for models.