fix references to models