diff --git a/example.py b/example.py
index 0daa6170..ec186341 100644
--- a/example.py
+++ b/example.py
@@ -27,7 +27,7 @@
 
 
 ########## V2
-from swarms import OpenAI
+from swarms.agents.models.openai import OpenAI
 
 chat = OpenAI()
 response = chat.generate("Hello World")