     imageClassifierOptions.classificationOptions.maxResults = maxResults
 
     let imageClassifier =
-      try TFLImageClassifier.imageClassifier(options: imageClassifierOptions)
+      try XCTUnwrap(TFLImageClassifier.imageClassifier(options: imageClassifierOptions))
 
     let gmlImage = try XCTUnwrap(
       MLImage.imageFromBundle(

    
    // TODO: match the score as image_classifier_test.cc
    self.verifyCategory(classificationResult.classifications[0].categories[0], 
                        expectedIndex: 934, 
                        expectedScore: 0.748976,
                        expectedLabel: "cheeseburger", 
                        expectedDisplayName: nil);
    self.verifyCategory(classificationResult.classifications[0].categories[1], 
                        expectedIndex: 925, 
                        expectedScore: 0.024646, 
                        expectedLabel: "guacamole",
                        expectedDisplayName: nil);
    self.verifyCategory(classificationResult.classifications[0].categories[2], 
                        expectedIndex: 932, 
                        expectedScore: 0.022505, 
                        expectedLabel: "bagel",
                        expectedDisplayName: nil);
  }

    // TODO: match the label and score as image_classifier_test.cc
    // let category = classificationResults.classifications[0].categories[0]
    // XCTAssertEqual(category.label, "soccer ball")
    // XCTAssertEqual(category.score, 0.256512, accuracy:0.001);
  }

  func testInferenceWithRGBAImage() throws {

        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], 
