Use updated models