diff --git a/scenarios/evaluate/Supported_Evaluation_Metrics/AI_Judge_Evaluators_Safety_Risks/AI_Judge_Evaluators_Safety_Risks_Audio/AI_Judge_Evaluators_Safety_Risks_Audio.ipynb b/scenarios/evaluate/Supported_Evaluation_Metrics/AI_Judge_Evaluators_Safety_Risks/AI_Judge_Evaluators_Safety_Risks_Audio/AI_Judge_Evaluators_Safety_Risks_Audio.ipynb
index a853d713..285665bc 100644
--- a/scenarios/evaluate/Supported_Evaluation_Metrics/AI_Judge_Evaluators_Safety_Risks/AI_Judge_Evaluators_Safety_Risks_Audio/AI_Judge_Evaluators_Safety_Risks_Audio.ipynb
+++ b/scenarios/evaluate/Supported_Evaluation_Metrics/AI_Judge_Evaluators_Safety_Risks/AI_Judge_Evaluators_Safety_Risks_Audio/AI_Judge_Evaluators_Safety_Risks_Audio.ipynb
@@ -345,10 +345,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from rtclient import RTInputItem\n",
+    "from rtclient import RTInputAudioItem\n",
     "\n",
     "\n",
-    "async def receive_input_item(start_time: float, item: RTInputItem) -> None:\n",
+    "async def receive_input_item(start_time: float, item: RTInputAudioItem) -> None:\n",
     "    prefix = f\"[input_item={item.id}]\"\n",
     "    await item\n",
     "    log(start_time, prefix, f\"Previous Id: {item.previous_id}\")\n",
@@ -484,7 +484,7 @@
     "outputs = await adv_simulator(\n",
     "    scenario=adv_scenario,\n",
     "    max_conversation_turns=3,  # define the number of conversation turns\n",
-    "    max_simulation_results=3,  # define the number of simulation results\n",
+    "    max_simulation_results=1,  # define the number of simulation results\n",
     "    target=audio_callback,  # define the target model callback\n",
     "    concurrent_async_task=1,\n",
     ")"
@@ -536,7 +536,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "audio",
+   "display_name": "audio-package-test",
    "language": "python",
    "name": "python3"
   },