diff --git a/src/documentation/document_index.ipynb b/src/documentation/document_index.ipynb
index e3ea71c9b..cbe22b86a 100644
--- a/src/documentation/document_index.ipynb
+++ b/src/documentation/document_index.ipynb
@@ -22,7 +22,7 @@
     "    LimitedConcurrencyClient,\n",
     ")\n",
     "from intelligence_layer.core import InMemoryTracer\n",
-    "from intelligence_layer.examples import RetrieverBasedQa, RetrieverBasedQaInput\n",
+    "from intelligence_layer.examples import MultipleChunkRetrieverQa, RetrieverBasedQaInput\n",
     "\n",
     "load_dotenv()"
    ]
@@ -266,7 +266,7 @@
     "\n",
     "Finally, since we have a ready-to use 'Retriever', we can employ it to do something more complicated than just search; it can serve as the basis for a question-answering system.\n",
     "\n",
-    "To do so, let's run a `RetrieverBasedQa` task:"
+    "To do so, let's run a `MultipleChunkRetrieverQa` task:"
    ]
   },
   {
@@ -276,7 +276,7 @@
    "outputs": [],
    "source": [
     "client = LimitedConcurrencyClient.from_env()\n",
-    "retriever_qa = RetrieverBasedQa(document_index_retriever)\n",
+    "retriever_qa = MultipleChunkRetrieverQa(document_index_retriever, insert_chunk_number=3)\n",
     "\n",
     "\n",
     "input = RetrieverBasedQaInput(\n",