diff --git a/notebooks/rag.ipynb b/notebooks/rag.ipynb
index 42adaed..6420d38 100644
--- a/notebooks/rag.ipynb
+++ b/notebooks/rag.ipynb
@@ -1547,11 +1547,11 @@
     "        contexts_results = semantic_search(\n",
     "            query=query, \n",
     "            embedding_model=self.embedding_model, \n",
-    "            num_chunks=num_chunks)\n",
+    "            k=num_chunks)\n",
     "            \n",
     "        # Generate response\n",
-    "        context = [item[\"text\"] for item in context_results]\n",
-    "        sources = [item[\"source\"] for item in context_results]\n",
+    "        context = [item[\"text\"] for item in contexts_results]\n",
+    "        sources = [item[\"source\"] for item in contexts_results]\n",
     "        user_content = f\"query: {query}, context: {context}\"\n",
     "        answer = generate_response(\n",
     "            llm=self.llm,\n",
@@ -7168,7 +7168,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.8"
+   "version": "3.11.4"
   }
  },
  "nbformat": 4,