         transcribeDataCustomForNumbers(
             filename: "StockAnnouncement",
             withExtension: "wav",
-            format: .wav,
+            format: "audio/wav",
             substring: "$152.37"
         )
     }
 
-    func transcribeDataCustomForNumbers(
-        filename: String,
-        withExtension: String,
-        format: AudioMediaType,
-        substring: String)
-    {
+    func transcribeDataCustomForNumbers(filename: String, withExtension: String, format: String, substring: String) {
         let description = "Transcribe an audio file with smart formatting."
         let expectation = self.expectation(description: description)
 

        let description1 = "Get information about all models."
        let expectation1 = expectation(description: description1)

        var allModels = [Model]()
        speechToText.getModels(failure: failWithError) { models in
            allModels = models
            expectation1.fulfill()
        }
        waitForExpectations()


        let description2 = "Train the customization with the new corpus."
        let expectation2 = self.expectation(description: description2)
        speechToText.trainLanguageModel(customizationID: self.trainedCustomizationID, failure: failToTrainCustomization) {
            expectation2.fulfill()
        }
        waitForExpectations()
