IMO it would be simpler if we just added some custom logic here like setting an attribute on the model to a certain value