correct some models tests for new document