return of model fix