diff --git a/docs/docs/integrations/llms/fireworks.ipynb b/docs/docs/integrations/llms/fireworks.ipynb
index 2aaae1b699c..57e772f760c 100644
--- a/docs/docs/integrations/llms/fireworks.ipynb
+++ b/docs/docs/integrations/llms/fireworks.ipynb
@@ -54,6 +54,8 @@
     "import getpass\n",
     "import os\n",
     "\n",
+    "from langchain_fireworks import Fireworks\n",
+    "\n",
     "if \"FIREWORKS_API_KEY\" not in os.environ:\n",
     "    os.environ[\"FIREWORKS_API_KEY\"] = getpass.getpass(\"Fireworks API Key:\")\n",
     "\n",
@@ -181,7 +183,7 @@
    ],
    "source": [
     "from langchain.prompts import PromptTemplate\n",
-    "from langchain_community.llms.fireworks import Fireworks\n",
+    "from langchain_fireworks import Fireworks\n",
     "\n",
     "llm = Fireworks(\n",
     "    model=\"accounts/fireworks/models/mixtral-8x7b-instruct\",\n",
@@ -249,7 +251,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.4"
+   "version": "3.9.6"
   }
  },
  "nbformat": 4,