diff --git a/docs/modules/prompts/examples/custom_prompt_template.md b/docs/modules/prompts/examples/custom_prompt_template.md
index c3358eb0048fe..4caa6131c1ee3 100644
--- a/docs/modules/prompts/examples/custom_prompt_template.md
+++ b/docs/modules/prompts/examples/custom_prompt_template.md
@@ -34,7 +34,7 @@ Next, we'll create a custom prompt template that takes in the function name as i
 
 ```python
 from langchain.prompts import BasePromptTemplate
-from pydantic import BaseModel
+from pydantic import BaseModel, validator
 
 
 class FunctionExplainerPromptTemplate(BasePromptTemplate, BaseModel):