diff --git a/docs/snippets/modules/agents/agent_types/openai_functions_agent.mdx b/docs/snippets/modules/agents/agent_types/openai_functions_agent.mdx
index a04e2ac8c7..aaf208dc05 100644
--- a/docs/snippets/modules/agents/agent_types/openai_functions_agent.mdx
+++ b/docs/snippets/modules/agents/agent_types/openai_functions_agent.mdx
@@ -5,10 +5,12 @@ pip install openai google-search-results
 ```
 
 ```python
-from langchain import LLMMathChain, OpenAI, SerpAPIWrapper, SQLDatabase, SQLDatabaseChain
-from langchain.agents import initialize_agent, Tool
-from langchain.agents import AgentType
+from langchain.agents import initialize_agent, AgentType, Tool
+from langchain.chains import LLMMathChain
 from langchain.chat_models import ChatOpenAI
+from langchain.llms import OpenAI
+from langchain.utilities import SerpAPIWrapper, SQLDatabase
+from langchain_experimental.sql import SQLDatabaseChain
 ```