diff --git a/docs/scripts/model_feat_table.py b/docs/scripts/model_feat_table.py
index 7f4fa0ca1d..ecf37fbcd6 100644
--- a/docs/scripts/model_feat_table.py
+++ b/docs/scripts/model_feat_table.py
@@ -1,7 +1,7 @@
 import os
 from pathlib import Path
 
-from langchain import chat_models, llms
+from langchain_community import chat_models, llms
 from langchain_core.language_models.chat_models import BaseChatModel, SimpleChatModel
 from langchain_core.language_models.llms import LLM, BaseLLM