model fix pr.