diff --git a/agents-and-function-calling/bedrock-agents/features-examples/05-create-agent-with-knowledge-base-and-action-group/05-create-agent-with-knowledge-base-and-action-group.ipynb b/agents-and-function-calling/bedrock-agents/features-examples/05-create-agent-with-knowledge-base-and-action-group/05-create-agent-with-knowledge-base-and-action-group.ipynb
index 3999856cb..c09798ed9 100644
--- a/agents-and-function-calling/bedrock-agents/features-examples/05-create-agent-with-knowledge-base-and-action-group/05-create-agent-with-knowledge-base-and-action-group.ipynb
+++ b/agents-and-function-calling/bedrock-agents/features-examples/05-create-agent-with-knowledge-base-and-action-group/05-create-agent-with-knowledge-base-and-action-group.ipynb
@@ -1176,26 +1176,31 @@
     "        endSession= end_session,\n",
     "        sessionState=session_state\n",
     "    )\n",
-    "    \n",
+    "  \n",
     "    if enable_trace:\n",
     "        logger.info(pprint.pprint(agentResponse))\n",
     "    \n",
     "    event_stream = agentResponse['completion']\n",
     "    try:\n",
+    "        agent_answer = \"\"\n",
     "        for event in event_stream:        \n",
     "            if 'chunk' in event:\n",
     "                data = event['chunk']['bytes']\n",
+    "                chunk_text = data.decode('utf8')\n",
+    "                agent_answer += chunk_text\n",
+    "\n",
     "                if enable_trace:\n",
-    "                    logger.info(f\"Final answer ->\\n{data.decode('utf8')}\")\n",
-    "                agent_answer = data.decode('utf8')\n",
-    "                end_event_received = True\n",
-    "                return agent_answer\n",
-    "                # End event indicates that the request finished successfully\n",
+    "                    logger.info(f\"Chunk received ->\\n{chunk_text}\")\n",
     "            elif 'trace' in event:\n",
     "                if enable_trace:\n",
     "                    logger.info(json.dumps(event['trace'], indent=2, cls=DateTimeEncoder))\n",
     "            else:\n",
     "                raise Exception(\"unexpected event.\", event)\n",
+    "\n",
+    "            if enable_trace:\n",
+    "                logger.info(f\"Final answer ->\\n{agent_answer}\")\n",
+    "            \n",
+    "            return agent_answer\n",
     "    except Exception as e:\n",
     "        raise Exception(\"unexpected event.\", e)"
    ]