diff --git a/tests/hermes_cli/test_xai_retirement.py b/tests/hermes_cli/test_xai_retirement.py
index c1ad77bd8f..90d7442360 100644
--- a/tests/hermes_cli/test_xai_retirement.py
+++ b/tests/hermes_cli/test_xai_retirement.py
@@ -5,6 +5,7 @@ from __future__ import annotations
 import yaml
 
 from hermes_cli.xai_retirement import (
+    RetirementIssue,
     _RETIRED_MODELS,
     _looks_like_xai,
     _normalize,