Fix models tests.