Fixed model search bug