fixed broken model references