diff --git a/langchain/llms/ai21.py b/langchain/llms/ai21.py
index e0b804c8bb7..169718e26d1 100644
--- a/langchain/llms/ai21.py
+++ b/langchain/llms/ai21.py
@@ -28,7 +28,7 @@ class AI21(LLM, BaseModel):
     Example:
         .. code-block:: python
 
-            from langchain import AI21
+            from langchain.llms import AI21
             ai21 = AI21(model="j1-jumbo")
     """