diff --git a/docs/extras/integrations/tools/awslambda.ipynb b/docs/extras/integrations/tools/awslambda.ipynb
index 6f6f8e9fef..7bc492bb6a 100644
--- a/docs/extras/integrations/tools/awslambda.ipynb
+++ b/docs/extras/integrations/tools/awslambda.ipynb
@@ -66,7 +66,7 @@
    "outputs": [],
    "source": [
     "from langchain import OpenAI\n",
-    "from langchain.agents import load_tools, AgentType\n",
+    "from langchain.agents import load_tools, initialize_agent, AgentType\n",
     "\n",
     "llm = OpenAI(temperature=0)\n",
     "\n",