diff --git a/__init__.py b/__init__.py
index 3254229..eec8e9c 100644
--- a/__init__.py
+++ b/__init__.py
@@ -12,9 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from .image_decode_cv2 import ImageDecodeCV2
+from .image_decode_cv2 import ImageDecoder
 
 
 # The factory method
 def test_cv2():
-    return ImageDecodeCV2()
+    return ImageDecoder()
diff --git a/image_decode_cv2.py b/image_decode_cv2.py
index b6a6012..70bcc37 100644
--- a/image_decode_cv2.py
+++ b/image_decode_cv2.py
@@ -6,7 +6,7 @@ import cv2
 import requests
 import numpy as np
 
-from towhee.types import Image
+from towhee._types import Image
 from towhee.operator import PyOperator, SharedType
 from towhee import register
 from towhee.utils import ndarray_utils