minor change on model def