fix tests

Author: radofuchs

tests/e2e/features/steps/info.py

@@ -82,10 +82,10 @@ def check_shield_structure(context: Context) -> None:
             found_shield = shield
             break
 
-    assert found_shield is not None, "No LLM model found in response"
+    assert found_shield is not None, "No shield found in response"
 
     # Validate structure and values
-    assert found_shield["type"] == "shield", "type should be 'model'"
+    assert found_shield["type"] == "shield", "type should be 'shield'"
     assert (
         found_shield["provider_id"] == "llama-guard"
     ), "provider_id should be 'llama-guard'"