Update to the model to fix a unit test in admin