fix test broken because of wrong model fix