diff --git a/mediapipe/model_maker/__init__.py b/mediapipe/model_maker/__init__.py
index 6779524b2..d43536663 100644
--- a/mediapipe/model_maker/__init__.py
+++ b/mediapipe/model_maker/__init__.py
@@ -14,10 +14,13 @@
 
 
 from mediapipe.model_maker.python.core.utils import quantization
+from mediapipe.model_maker.python.core.utils import model_util
+
 from mediapipe.model_maker.python.vision import image_classifier
 from mediapipe.model_maker.python.vision import gesture_recognizer
 from mediapipe.model_maker.python.text import text_classifier
 from mediapipe.model_maker.python.vision import object_detector
+from mediapipe.model_maker.python.vision import face_stylizer
 
 # Remove duplicated and non-public API
 del python