Use different model for add and assign