Cleanup imports in model.