Fix returning model