small fix to return model on update