update models tests to new API