diff --git a/tests/e2e/test_analyze_e2e.py b/tests/e2e/test_analyze_e2e.py
index babcf38fb..343ba32a1 100644
--- a/tests/e2e/test_analyze_e2e.py
+++ b/tests/e2e/test_analyze_e2e.py
@@ -201,7 +201,7 @@ def test_invalid_ep_choice_exits_two(self, onnx_model_path: Path) -> None:
     def test_invalid_device_choice_exits_two(self, onnx_model_path: Path) -> None:
         result = _invoke(["-m", str(onnx_model_path), "--device", "TPU"])
         assert result.exit_code == 2
-        assert "Invalid value for '--device'" in result.output
+        assert "Invalid value for '-d' / '--device'" in result.output
 
     def test_invalid_save_node_choice_exits_two(self, onnx_model_path: Path) -> None:
         result = _invoke(["-m", str(onnx_model_path), "--save-node", "supported"])