Fixed model lookup error