diff --git a/steps/src/toxicity_guardrail/toxicity_guardrail.ipynb b/steps/src/toxicity_guardrail/toxicity_guardrail.ipynb
index ddc3fe99..be9d6725 100644
--- a/steps/src/toxicity_guardrail/toxicity_guardrail.ipynb
+++ b/steps/src/toxicity_guardrail/toxicity_guardrail.ipynb
@@ -200,7 +200,7 @@
     ")\n",
     "\n",
     "model_runner = ModelRunnerStep(name=\"llm_runner\")\n",
-    "model_runner.add_model(endpoint_name=\"llm_model\", model_class=\"LLMModel\")\n",
+    "model_runner.add_model(endpoint_name=\"llm_model\", model_class=\"LLMModel\", execution_mechanism=\"naive\")\n",
     "graph.add_step(model_runner, after=\"toxicity_guardrail\")\n",
     "\n",
     "graph.add_step(name=\"format_answer\", handler=\"format_answer\", after=\"llm_runner\").respond()\n",