       expectedClassificationsCount)
   }
 
-  func testSuccessfullInferenceOnMLImageWithUIImage() throws {
+  func testInferenceOnMLImageWithUIImage() throws {
 
     let modelPath = try XCTUnwrap(TFLImageClassifierTests.modelPath)
 
     let imageClassifierOptions = try XCTUnwrap(TFLImageClassifierOptions(modelPath: modelPath))
 
     let imageClassifier =
-      try TFLImageClassifier.imageClassifier(options: imageClassifierOptions)
+      try XCTUnwrap(TFLImageClassifier.imageClassifier(options: imageClassifierOptions))
 
     let gmlImage = try XCTUnwrap(
       MLImage.imageFromBundle(

                        expectedDisplayName: nil);
  }

  func testModelOptionsWithMaxResults() throws {

    let modelPath = try XCTUnwrap(TFLImageClassifierTests.modelPath)

        filename: "burger",
        type: "jpg"))

    let classificationResult = 
      try XCTUnwrap(imageClassifier.classify(gmlImage: gmlImage))

    let expectedClassificationsCount = 1
    self.verifyClassificationResult(classificationResult, 
                                    expectedClassificationsCount: expectedClassificationsCount)

    let expectedHeadIndex = 0
    self.verifyClassifications(classificationResult.classifications[0], 
