diff --git a/cookbook/LiteLLM_CometAPI.ipynb b/cookbook/LiteLLM_CometAPI.ipynb
index ceab0f10683..acf7f2abcf0 100644
--- a/cookbook/LiteLLM_CometAPI.ipynb
+++ b/cookbook/LiteLLM_CometAPI.ipynb
@@ -309,8 +309,7 @@
    "outputs": [],
    "source": [
     "moderation_response = litellm.moderation(\n",
-    "    model=\"omni-moderation-latest\",\n",
-    "    custom_llm_provider=\"cometapi\",\n",
+    "    model=\"cometapi/omni-moderation-latest\",\n",
     "    input=\"I want to build a safe application.\",\n",
     "    api_key=api_key,\n",
     ")\n",
@@ -318,6 +317,13 @@
     "print(moderation_response.results[0].categories)\n",
     "print(moderation_response.results[0].category_scores)\n"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {