diff --git a/libs/prebuilt/tests/test_react_agent.py b/libs/prebuilt/tests/test_react_agent.py
index 05aa833f3..4f91d4610 100644
--- a/libs/prebuilt/tests/test_react_agent.py
+++ b/libs/prebuilt/tests/test_react_agent.py
@@ -2035,7 +2035,7 @@ def test_post_model_hook_with_structured_output() -> None:
     events = list(
         agent.stream({"messages": [HumanMessage("What's the weather?")], "flag": False})
     )
-    assert "generate_structured_response" in events[-1]
+    assert "respond" in events[-1]
     assert events == [
         {
             "agent": {
@@ -2158,3 +2158,4 @@ def test_create_react_agent_inject_vars_with_post_model_hook(
     ]
     assert result["foo"] == 2
 
+