diff --git a/mediapipe/tasks/python/test/vision/face_stylizer_test.py b/mediapipe/tasks/python/test/vision/face_stylizer_test.py
index 3c39851dd..32643821f 100644
--- a/mediapipe/tasks/python/test/vision/face_stylizer_test.py
+++ b/mediapipe/tasks/python/test/vision/face_stylizer_test.py
@@ -36,7 +36,7 @@ _FaceStylizerOptions = face_stylizer.FaceStylizerOptions
 _RUNNING_MODE = running_mode_module.VisionTaskRunningMode
 _ImageProcessingOptions = image_processing_options_module.ImageProcessingOptions
 
-_MODEL = 'face_stylizer_model_placeholder.tflite'
+_MODEL = 'face_stylization_dummy.tflite'
 _IMAGE = 'cats_and_dogs.jpg'
 _STYLIZED_IMAGE = 'stylized_image_placeholder.jpg'
 _TEST_DATA_DIR = 'mediapipe/tasks/testdata/vision'
@@ -104,11 +104,9 @@ class FaceStylizerTest(parameterized.TestCase):
     stylizer = _FaceStylizer.create_from_options(options)
 
     # Performs face stylization on the input.
-    stylized_image = stylizer.detect(self.test_image)
+    stylized_image = stylizer.stylize(self.test_image)
     # Comparing results.
-    self.assertTrue(
-      np.array_equal(stylized_image.numpy_view(),
-                     self.test_image.numpy_view()))
+    # TODO:
     # Closes the stylizer explicitly when the stylizer is not used in
     # a context.
     stylizer.close()
diff --git a/mediapipe/tasks/python/vision/BUILD b/mediapipe/tasks/python/vision/BUILD
index e21171fc2..f89ef04da 100644
--- a/mediapipe/tasks/python/vision/BUILD
+++ b/mediapipe/tasks/python/vision/BUILD
@@ -162,7 +162,7 @@ py_library(
         "//mediapipe/python:_framework_bindings",
         "//mediapipe/python:packet_creator",
         "//mediapipe/python:packet_getter",
-        "//mediapipe/tasks/cc/vision/image_segmenter/proto:face_stylizer_graph_options_py_pb2",
+        "//mediapipe/tasks/cc/vision/face_stylizer/proto:face_stylizer_graph_options_py_pb2",
         "//mediapipe/tasks/python/core:base_options",
         "//mediapipe/tasks/python/core:optional_dependencies",
         "//mediapipe/tasks/python/core:task_info",
diff --git a/mediapipe/tasks/python/vision/face_stylizer.py b/mediapipe/tasks/python/vision/face_stylizer.py
index cd840fe85..1393982da 100644
--- a/mediapipe/tasks/python/vision/face_stylizer.py
+++ b/mediapipe/tasks/python/vision/face_stylizer.py
@@ -124,8 +124,10 @@ class FaceStylizer(base_vision_task_api.BaseVisionTaskApi):
         return
       image = packet_getter.get_image(output_packets[_IMAGE_OUT_STREAM_NAME])
       stylized_image_packet = output_packets[_STYLIZED_IMAGE_NAME]
+      stylized_image = packet_getter.get_image(stylized_image_packet)
+
       options.result_callback(
-        stylized_image_packet, image,
+        stylized_image, image,
         stylized_image_packet.timestamp.value // _MICRO_SECONDS_PER_MILLISECOND)
 
     task_info = _TaskInfo(
@@ -173,7 +175,7 @@ class FaceStylizer(base_vision_task_api.BaseVisionTaskApi):
         _NORM_RECT_STREAM_NAME:
             packet_creator.create_proto(normalized_rect.to_pb2())
     })
-    return output_packets[_STYLIZED_IMAGE_NAME]
+    return packet_getter.get_image(output_packets[_STYLIZED_IMAGE_NAME])
 
   def stylize_for_video(
       self,
@@ -209,7 +211,7 @@ class FaceStylizer(base_vision_task_api.BaseVisionTaskApi):
             packet_creator.create_proto(normalized_rect.to_pb2()).at(
                 timestamp_ms * _MICRO_SECONDS_PER_MILLISECOND)
     })
-    return output_packets[_STYLIZED_IMAGE_NAME]
+    return packet_getter.get_image(output_packets[_STYLIZED_IMAGE_NAME])
 
   def stylize_async(
       self,
diff --git a/mediapipe/tasks/testdata/vision/BUILD b/mediapipe/tasks/testdata/vision/BUILD
index 097acad43..f77e1780c 100644
--- a/mediapipe/tasks/testdata/vision/BUILD
+++ b/mediapipe/tasks/testdata/vision/BUILD
@@ -170,6 +170,7 @@ filegroup(
         "face_detection_short_range.tflite",
         "face_landmark.tflite",
         "face_landmark_with_attention.tflite",
+        "face_stylization_dummy.tflite",
         "face_landmarker.task",
         "hair_segmentation.tflite",
         "hand_landmark_full.tflite",
diff --git a/mediapipe/tasks/testdata/vision/face_stylization_dummy.tflite b/mediapipe/tasks/testdata/vision/face_stylization_dummy.tflite
new file mode 100644
index 000000000..2523a3a5d
Binary files /dev/null and b/mediapipe/tasks/testdata/vision/face_stylization_dummy.tflite differ