Fix models reference issue.