diff --git a/__init__.py b/__init__.py
index a2ef089..a782706 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,5 +1,5 @@
-from .openai_chat import OpenAI
+from .openai_chat import OpenaiChat
 
 
 def OpenAI(*args, **kwargs):
-    return OpenAI(*args, **kwargs)
+    return OpenaiChat(*args, **kwargs)
diff --git a/openai_chat.py b/openai_chat.py
index 3dce8af..dc1850d 100644
--- a/openai_chat.py
+++ b/openai_chat.py
@@ -19,7 +19,7 @@ import openai
 from towhee.operator.base import PyOperator
 
 
-class OpenAI(PyOperator):
+class OpenaiChat(PyOperator):
     '''Wrapper of OpenAI Chat API'''
     def __init__(self,
                  model_name: str = 'gpt-3.5-turbo',