diff --git a/tests/test_guardrails.py b/tests/test_guardrails.py
index f863983b2f..9a7db63774 100644
--- a/tests/test_guardrails.py
+++ b/tests/test_guardrails.py
@@ -228,6 +228,7 @@ async def test_input_guardrail_decorators():
     )
     assert not result.output.tripwire_triggered
     assert result.output.output_info == "test_1"
+    assert guardrail.get_name() == "decorated_input_guardrail"
 
     guardrail = decorated_named_input_guardrail
     result = await guardrail.run(
@@ -266,6 +267,7 @@ async def test_output_guardrail_decorators():
     )
     assert not result.output.tripwire_triggered
     assert result.output.output_info == "test_3"
+    assert guardrail.get_name() == "decorated_output_guardrail"
 
     guardrail = decorated_named_output_guardrail
     result = await guardrail.run(
diff --git a/tests/test_tool_guardrails.py b/tests/test_tool_guardrails.py
index 8ccaec0ad6..30e862f1fa 100644
--- a/tests/test_tool_guardrails.py
+++ b/tests/test_tool_guardrails.py
@@ -261,6 +261,7 @@ async def test_tool_input_guardrail_decorators():
     result = await guardrail.run(data)
     assert result.behavior["type"] == "allow"
     assert result.output_info == "test_1"
+    assert guardrail.get_name() == "decorated_input_guardrail"
 
     # Test named decorator
     guardrail = decorated_named_input_guardrail
@@ -294,6 +295,7 @@ async def test_tool_output_guardrail_decorators():
     result = await guardrail.run(data)
     assert result.behavior["type"] == "allow"
     assert result.output_info == "test_3"
+    assert guardrail.get_name() == "decorated_output_guardrail"
 
     # Test named decorator
     guardrail = decorated_named_output_guardrail