diff --git a/docs/docs/integrations/chat/ai21.ipynb b/docs/docs/integrations/chat/ai21.ipynb
index b519548521d..b009662abc6 100644
--- a/docs/docs/integrations/chat/ai21.ipynb
+++ b/docs/docs/integrations/chat/ai21.ipynb
@@ -17,8 +17,6 @@
       "source": [
         "# ChatAI21\n",
         "\n",
-        "## Overview\n",
-        "\n",
         "This notebook covers how to get started with AI21 chat models.\n",
         "Note that different chat models support different parameters. See the [AI21 documentation](https://docs.ai21.com/reference) to learn more about the parameters in your chosen model.\n",
         "[See all AI21's LangChain components.](https://pypi.org/project/langchain-ai21/)\n",
@@ -68,7 +66,9 @@
       "cell_type": "markdown",
       "id": "f6844fff-3702-4489-ab74-732f69f3b9d7",
       "metadata": {},
-      "source": "To enable automated tracing of your model calls, set your [LangSmith](https://docs.smith.langchain.com/) API key:"
+      "source": [
+        "To enable automated tracing of your model calls, set your [LangSmith](https://docs.smith.langchain.com/) API key:"
+      ]
     },
     {
       "cell_type": "code",
@@ -198,13 +198,17 @@
       "cell_type": "markdown",
       "id": "39c0ccd229927eab",
       "metadata": {},
-      "source": "# Tool Calls / Function Calling"
+      "source": [
+        "# Tool Calls / Function Calling"
+      ]
     },
     {
       "cell_type": "markdown",
       "id": "2bf6b40be07fe2d4",
       "metadata": {},
-      "source": "This example shows how to use tool calling with AI21 models:"
+      "source": [
+        "This example shows how to use tool calling with AI21 models:"
+      ]
     },
     {
       "cell_type": "code",