diff --git a/notebooks/query_decomposition.ipynb b/notebooks/query_decomposition.ipynb
index 98cee06..5fd1223 100644
--- a/notebooks/query_decomposition.ipynb
+++ b/notebooks/query_decomposition.ipynb
@@ -56,6 +56,37 @@
         "from typing import List, Any, Dict, Optional, Callable, Union"
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "7JGz6wd6mZ2M",
+        "outputId": "9d577b8c-7aff-4604-f9b6-847ff2b5534a"
+      },
+      "outputs": [
+        {
+          "name": "stdout",
+          "output_type": "stream",
+          "text": [
+            "OpenAI API Key:··········\n",
+            "Cohere API Key:··········\n"
+          ]
+        }
+      ],
+      "source": [
+        "import os\n",
+        "from getpass import getpass\n",
+        "\n",
+        "if \"OPENAI_API_KEY\" not in os.environ:\n",
+        "  os.environ[\"OPENAI_API_KEY\"] = getpass(\"OpenAI API Key:\")\n",
+        "\n",
+        "if \"COHERE_API_KEY\" not in os.environ:\n",
+        "  os.environ[\"COHERE_API_KEY\"] = getpass(\"Cohere API Key:\")"
+      ]
+    },
     {
       "cell_type": "markdown",
       "metadata": {
@@ -173,37 +204,6 @@
         "        return chat_message"
       ]
     },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "colab": {
-          "base_uri": "https://localhost:8080/"
-        },
-        "id": "7JGz6wd6mZ2M",
-        "outputId": "9d577b8c-7aff-4604-f9b6-847ff2b5534a"
-      },
-      "outputs": [
-        {
-          "name": "stdout",
-          "output_type": "stream",
-          "text": [
-            "OpenAI API Key:··········\n",
-            "Cohere API Key:··········\n"
-          ]
-        }
-      ],
-      "source": [
-        "import os\n",
-        "from getpass import getpass\n",
-        "\n",
-        "if \"OPENAI_API_KEY\" not in os.environ:\n",
-        "  os.environ[\"OPENAI_API_KEY\"] = getpass(\"OpenAI API Key:\")\n",
-        "\n",
-        "if \"COHERE_API_KEY\" not in os.environ:\n",
-        "  os.environ[\"COHERE_API_KEY\"] = getpass(\"Cohere API Key:\")"
-      ]
-    },
     {
       "cell_type": "markdown",
       "metadata": {