diff --git a/openai_qa.py b/openai_qa.py
index 1aafcc1..7ac3469 100644
--- a/openai_qa.py
+++ b/openai_qa.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 
-from openai import Embedding
+import openai
 from tenacity import retry, stop_after_attempt, wait_random_exponential
 from towhee.operator.base import PyOperator