fix models tests