diff --git a/docs/extras/modules/model_io/models/llms/integrations/huggingface_textgen_inference.ipynb b/docs/extras/modules/model_io/models/llms/integrations/huggingface_textgen_inference.ipynb
index 8fa72d948b..72c79cee7b 100644
--- a/docs/extras/modules/model_io/models/llms/integrations/huggingface_textgen_inference.ipynb
+++ b/docs/extras/modules/model_io/models/llms/integrations/huggingface_textgen_inference.ipynb
@@ -35,6 +35,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from langchain.llms import HuggingFaceTextGenInference\n",
+    "\n",
     "llm = HuggingFaceTextGenInference(\n",
     "    inference_server_url=\"http://localhost:8010/\",\n",
     "    max_new_tokens=512,\n",