     let imageClassifier =
       try TFLImageClassifier.imageClassifier(options: imageClassifierOptions!)
     
-    let imagePath = try XCTUnwrap(TFLImageClassifierTests.bundle.path(forResource: "burger_crop", ofType: "jpg"))
-    let image = UIImage(contentsOfFile: imagePath)
-    let imageForInference = try XCTUnwrap(image)
-    let gmlImage =  try XCTUnwrap(MLImage(image: imageForInference))
+    let gmlImage = try gmlImage(withName:"burger_crop", ofType:"jpg");
     
     let classificationResults: TFLClassificationResult = try imageClassifier.classify(gmlImage:gmlImage)
     

    let imageClassifier =
      try TFLImageClassifier.imageClassifier(options: imageClassifierOptions!)
    
    let imagePath = try XCTUnwrap(TFLImageClassifierTests.bundle.path(forResource: "burger_crop", ofType: "jpg"))
    let image = UIImage(contentsOfFile: imagePath)
    let imageForInference = try XCTUnwrap(image)
    let gmlImage =  try XCTUnwrap(MLImage(image: imageForInference))
    
    let classificationResults =
      try imageClassifier.classify(gmlImage: gmlImage)

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

    let gmlImage = try gmlImage(withName:"burger_crop", ofType:"jpg");
    let classificationResults: TFLClassificationResult = try imageClassifier.classify(gmlImage:gmlImage)
      
    XCTAssertNotNil(classificationResults)
