update model to have right args