diff --git a/mediapipe/model_maker/python/vision/face_stylizer/dataset_test.py b/mediapipe/model_maker/python/vision/face_stylizer/dataset_test.py
index cf0dce800..73140f30e 100644
--- a/mediapipe/model_maker/python/vision/face_stylizer/dataset_test.py
+++ b/mediapipe/model_maker/python/vision/face_stylizer/dataset_test.py
@@ -22,8 +22,7 @@ class DatasetTest(tf.test.TestCase):
 
   def setUp(self):
     super().setUp()
-    # TODO: Replace the stylize image dataset with licensed images.
-    self._test_data_dirname = 'testdata'
+    self._test_data_dirname = 'input/style'
 
   def test_from_folder(self):
     input_data_dir = test_utils.get_test_data_path(self._test_data_dirname)
diff --git a/mediapipe/model_maker/python/vision/face_stylizer/face_stylizer_test.py b/mediapipe/model_maker/python/vision/face_stylizer/face_stylizer_test.py
index 16b314c8e..354ce7996 100644
--- a/mediapipe/model_maker/python/vision/face_stylizer/face_stylizer_test.py
+++ b/mediapipe/model_maker/python/vision/face_stylizer/face_stylizer_test.py
@@ -25,7 +25,7 @@ class FaceStylizerTest(tf.test.TestCase):
 
   def _load_data(self):
     """Loads training dataset."""
-    input_data_dir = test_utils.get_test_data_path('testdata')
+    input_data_dir = test_utils.get_test_data_path('input/style')
 
     data = face_stylizer.Dataset.from_folder(dirname=input_data_dir)
     return data
diff --git a/mediapipe/model_maker/python/vision/face_stylizer/testdata/cartoon/disney.png b/mediapipe/model_maker/python/vision/face_stylizer/testdata/cartoon/disney.png
deleted file mode 100644
index 87e9d3d8d..000000000
Binary files a/mediapipe/model_maker/python/vision/face_stylizer/testdata/cartoon/disney.png and /dev/null differ
diff --git a/mediapipe/model_maker/python/vision/face_stylizer/testdata/input/style/cartoon/cartoon.jpg b/mediapipe/model_maker/python/vision/face_stylizer/testdata/input/style/cartoon/cartoon.jpg
new file mode 100644
index 000000000..34f2d4c4a
Binary files /dev/null and b/mediapipe/model_maker/python/vision/face_stylizer/testdata/input/style/cartoon/cartoon.jpg differ
diff --git a/mediapipe/model_maker/python/vision/face_stylizer/testdata/input/style/sketch/sketch.jpg b/mediapipe/model_maker/python/vision/face_stylizer/testdata/input/style/sketch/sketch.jpg
new file mode 100644
index 000000000..477be0bdb
Binary files /dev/null and b/mediapipe/model_maker/python/vision/face_stylizer/testdata/input/style/sketch/sketch.jpg differ
diff --git a/mediapipe/model_maker/python/vision/face_stylizer/testdata/sketch/sketch.png b/mediapipe/model_maker/python/vision/face_stylizer/testdata/sketch/sketch.png
deleted file mode 100644
index 169192c96..000000000
Binary files a/mediapipe/model_maker/python/vision/face_stylizer/testdata/sketch/sketch.png and /dev/null differ