diff --git a/mediapipe/python/pybind/image.cc b/mediapipe/python/pybind/image.cc
index 800e883b4..62437e439 100644
--- a/mediapipe/python/pybind/image.cc
+++ b/mediapipe/python/pybind/image.cc
@@ -52,9 +52,9 @@ void ImageSubmodule(pybind11::module* module) {
   ```python
   import cv2
   cv_mat = cv2.imread(input_file)[:, :, ::-1]
-  rgb_frame = mp.Image(image_format=ImageFormat.SRGB, data=cv_mat)
+  rgb_frame = mp.Image(image_format=mp.ImageFormat.SRGB, data=cv_mat)
   gray_frame = mp.Image(
-      image_format=ImageFormat.GRAY,
+      image_format=mp.ImageFormat.GRAY8,
       data=cv2.cvtColor(cv_mat, cv2.COLOR_RGB2GRAY))
 
   from PIL import Image