       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(

    let imageClassifierOptions = try XCTUnwrap(TFLImageClassifierOptions(modelPath: modelPath))

    let imageClassifier =
      try TFLImageClassifier.imageClassifier(options: imageClassifierOptions)

    let gmlImage = try XCTUnwrap(
      MLImage.imageFromBundle(
        class: type(of: self),
        filename: "sparrow",
        type: "png"))

    let classificationResults =
      try imageClassifier.classify(gmlImage: gmlImage)

    XCTAssertNotNil(classificationResults)
    XCTAssertEqual(classificationResults.classifications.count, 1)
    XCTAssertGreaterThan(classificationResults.classifications[0].categories.count, 0)

    let category = classificationResults.classifications[0].categories[0]
    XCTAssertEqual(category.label, "junco")
    XCTAssertEqual(category.score, 0.253016, accuracy: 0.001)
  }
}

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

    let expectedCategoryCount = 1001
    let expectedHeadIndex = 0
