I think we can do this on the model