diff --git a/integrations/aws-strands/python/pyproject.toml b/integrations/aws-strands/python/pyproject.toml
index 66bd0fbe0..db84af55a 100644
--- a/integrations/aws-strands/python/pyproject.toml
+++ b/integrations/aws-strands/python/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "ag_ui_strands"
-version = "0.1.8"
+version = "0.1.9"
 authors = [
     { name = "AG-UI Contributors" }
 ]
diff --git a/integrations/aws-strands/python/tests/test_context_forwarding.py b/integrations/aws-strands/python/tests/test_context_forwarding.py
index b0ada4f7c..8ecdb65c4 100644
--- a/integrations/aws-strands/python/tests/test_context_forwarding.py
+++ b/integrations/aws-strands/python/tests/test_context_forwarding.py
@@ -10,6 +10,7 @@
 
 import pytest
 from strands import Agent
+from strands.agent.state import AgentState
 from strands.tools.registry import ToolRegistry
 
 from ag_ui.core import Context, RunAgentInput, UserMessage
@@ -38,7 +39,7 @@ class _CapturingCore:
     def __init__(self, **kwargs):
         self.init_kwargs = kwargs
         self.tool_registry = ToolRegistry()
-        self.state = JSONSerializableDict()
+        self.state = AgentState()
 
     async def stream_async(self, _msg: str):
         if False:
diff --git a/integrations/aws-strands/python/uv.lock b/integrations/aws-strands/python/uv.lock
index 71c3fffd4..80ec192e4 100644
--- a/integrations/aws-strands/python/uv.lock
+++ b/integrations/aws-strands/python/uv.lock
@@ -16,7 +16,7 @@ wheels = [
 
 [[package]]
 name = "ag-ui-strands"
-version = "0.1.8"
+version = "0.1.9"
 source = { editable = "." }
 dependencies = [
     { name = "ag-ui-protocol" },