diff --git a/notebooks/llmu/RAG_with_Chat_Embed_and_Rerank.ipynb b/notebooks/llmu/RAG_with_Chat_Embed_and_Rerank.ipynb
index ec7b98a..8aeb385 100644
--- a/notebooks/llmu/RAG_with_Chat_Embed_and_Rerank.ipynb
+++ b/notebooks/llmu/RAG_with_Chat_Embed_and_Rerank.ipynb
@@ -119,7 +119,7 @@
         "\n",
         "from IPython.display import HTML, display\n",
         "\n",
-        "def set_css():\n",
+        "def set_css(*args, **kwargs):\n",
         "  display(HTML('''\n",
         "  <style>\n",
         "    pre {\n",