         let description = "Retrieve a list of custom language models associated with this account."
         let expectation = self.expectation(description: description)
 
-        speechToText.getCustomizations(failure: failWithError) { customizations in
+        speechToText.listLanguageModels(failure: failWithError) { languageModels in
+            let customizations = languageModels.customizations
             XCTAssertGreaterThanOrEqual(customizations.count, 1)
-
             for customization in customizations {
                 XCTAssertNotNil(customization.customizationID)
                 XCTAssertNotNil(customization.created)

    // MARK: - Transcribe File, Custom Settings

    func testTranscribeFileCustomWAV() {
        transcribeFileCustom(filename: "SpeechSample", withExtension: "wav", format: .wav)
    }

    func testTranscribeFileCustomOpus() {
        transcribeFileCustom(filename: "SpeechSample", withExtension: "ogg", format: .oggOpus)
    }

    func testTranscribeFileCustomFLAC() {
        transcribeFileCustom(filename: "SpeechSample", withExtension: "flac", format: .flac)
    }

    func transcribeFileCustom(
        filename: String,
        withExtension: String,
        format: AudioMediaType)
    {
        let description = "Transcribe an audio file."
        let expectation = self.expectation(description: description)


    func testGetCorpusForTrainedCustomization() {
        let description = "Get the corpus used to build the trained customization."
        let expectation = self.expectation(description: description)
        speechToText.getCorpus(customizationID: trainedCustomizationID, corpusName: corpusName, failure: failWithError) {
            corpus in
            XCTAssertEqual(corpus.name, self.corpusName)
            XCTAssertEqual(corpus.status, "analyzed")
            XCTAssertEqual(corpus.totalWords, 40286)
            XCTAssertEqual(corpus.outOfVocabularyWords, 286)
            XCTAssertNil(corpus.error)
