diff --git a/mediapipe/tasks/python/test/vision/face_stylizer_test.py b/mediapipe/tasks/python/test/vision/face_stylizer_test.py
index c104d6a8c..c57618801 100644
--- a/mediapipe/tasks/python/test/vision/face_stylizer_test.py
+++ b/mediapipe/tasks/python/test/vision/face_stylizer_test.py
@@ -180,9 +180,7 @@ class FaceStylizerTest(parameterized.TestCase):
       image_processing_options = _ImageProcessingOptions(roi)
       # Performs face stylization on the input.
       stylized_image = stylizer.stylize(test_image, image_processing_options)
-      self.assertIsInstance(stylized_image, _Image)
-      self.assertEqual(stylized_image.width, _MODEL_IMAGE_SIZE)
-      self.assertEqual(stylized_image.height, _MODEL_IMAGE_SIZE)
+      self.assertIsNone(stylized_image)
 
   def test_missing_result_callback(self):
     options = _FaceStylizerOptions(