diff --git a/tests/kernels/attention/test_attention_selector.py b/tests/kernels/attention/test_attention_selector.py
index 4d969cf992d23..190c92e1251c2 100644
--- a/tests/kernels/attention/test_attention_selector.py
+++ b/tests/kernels/attention/test_attention_selector.py
@@ -368,4 +368,4 @@ def test_invalid_env(use_v1: bool, monkeypatch: pytest.MonkeyPatch):
         # Should raise ValueError for invalid backend
         with pytest.raises(ValueError) as exc_info:
             get_attn_backend(32, torch.float16, None, 16, False)
-        assert "Invalid attention backend: 'INVALID'" in str(exc_info.value)
+        assert "Invalid value 'INVALID'" in str(exc_info.value)