diff --git a/examples/chatgpt_agent/README.md b/examples/chatgpt_agent/README.md
index 42d693e67..c8e6c08a1 100644
--- a/examples/chatgpt_agent/README.md
+++ b/examples/chatgpt_agent/README.md
@@ -9,7 +9,7 @@ To install the ChatGPT agent, run the following command:
 ```{eval-rst}
 .. prompt:: bash
 
-    pip install flytekitplugins-chatgpt
+    pip install flytekitplugins-openai
 ```
 
 ## Example usage
diff --git a/examples/chatgpt_agent/chatgpt_agent/chatgpt_agent_example_usage.py b/examples/chatgpt_agent/chatgpt_agent/chatgpt_agent_example_usage.py
index 1a759e62d..2f2702939 100644
--- a/examples/chatgpt_agent/chatgpt_agent/chatgpt_agent_example_usage.py
+++ b/examples/chatgpt_agent/chatgpt_agent/chatgpt_agent_example_usage.py
@@ -12,7 +12,7 @@
 
 import flytekit
 from flytekit import ImageSpec, Secret, dynamic, task, workflow
-from flytekitplugins.chatgpt import ChatGPTTask
+from flytekitplugins.openai import ChatGPTTask
 
 # %% [markdown]
 # You have to specify your `name`, `openai_organization` and `chatgpt_config`.