diff --git a/python/src/tests/agents/test_amazon_bedrock_agent.py b/python/src/tests/agents/test_amazon_bedrock_agent.py
index d34b658..0c62873 100644
--- a/python/src/tests/agents/test_amazon_bedrock_agent.py
+++ b/python/src/tests/agents/test_amazon_bedrock_agent.py
@@ -50,7 +50,8 @@ async def test_process_request_success(bedrock_agent):
         agentId='test_agent_id',
         agentAliasId='test_agent_alias_id',
         sessionId='test_session',
-        inputText='Test input'
+        inputText='Test input',
+        sessionState={}
     )
 
 @pytest.mark.asyncio
@@ -116,4 +117,4 @@ async def test_process_request_with_additional_params(bedrock_agent):
 
     assert isinstance(result, ConversationMessage)
     assert result.role == ParticipantRole.ASSISTANT.value
-    assert result.content == [{"text": "Response with additional params"}]
\ No newline at end of file
+    assert result.content == [{"text": "Response with additional params"}]