diff --git a/mediapipe/python/BUILD b/mediapipe/python/BUILD
index 331ee836..55c0a22e 100644
--- a/mediapipe/python/BUILD
+++ b/mediapipe/python/BUILD
@@ -86,7 +86,6 @@ cc_library(
     name = "builtin_task_graphs",
     deps = [
         "//mediapipe/tasks/cc/vision/object_detector:object_detector_graph",
-        "//mediapipe/tasks/cc/vision/image_classification:image_classifier_graph",
         "//mediapipe/tasks/cc/vision/image_segmenter:image_segmenter_graph",
     ],
 )
diff --git a/mediapipe/tasks/python/test/vision/image_segmenter_test.py b/mediapipe/tasks/python/test/vision/image_segmenter_test.py
index 704b0fc5..6194f766 100644
--- a/mediapipe/tasks/python/test/vision/image_segmenter_test.py
+++ b/mediapipe/tasks/python/test/vision/image_segmenter_test.py
@@ -104,10 +104,11 @@ class ImageSegmenterTest(parameterized.TestCase):
     segmenter = _ImageSegmenter.create_from_options(options)
 
     # Performs image segmentation on the input.
-    image_result = segmenter.segment(self.test_image)
+    category_masks = segmenter.segment(self.test_image)
 
     # Comparing results.
-    print(image_result)
+    print(len(category_masks))
+    s
 
     # Closes the segmenter explicitly when the segmenter is not used in
     # a context.