Fix models reference