diff --git a/src/oss/langchain/guardrails.mdx b/src/oss/langchain/guardrails.mdx
index 34fef04a72..46ca584f6c 100644
--- a/src/oss/langchain/guardrails.mdx
+++ b/src/oss/langchain/guardrails.mdx
@@ -484,13 +484,7 @@ class SafetyGuardrailMiddleware(AgentMiddleware):
         result = self.safety_model.invoke([{"role": "user", "content": safety_prompt}])
 
         if "UNSAFE" in result.content:
-            return {
-                "messages": [{
-                    "role": "assistant",
-                    "content": "I cannot provide that response. Please rephrase your request."
-                }],
-                "jump_to": "end"
-            }
+            last_message.content = "I cannot provide that response. Please rephrase your request."
 
         return None
 
@@ -537,13 +531,7 @@ def safety_guardrail(state: AgentState, runtime: Runtime) -> dict[str, Any] | No
     result = safety_model.invoke([{"role": "user", "content": safety_prompt}])
 
     if "UNSAFE" in result.content:
-        return {
-            "messages": [{
-                "role": "assistant",
-                "content": "I cannot provide that response. Please rephrase your request."
-            }],
-            "jump_to": "end"
-        }
+        last_message.content = "I cannot provide that response. Please rephrase your request."
 
     return None