diff --git a/tests/test_cli.py b/tests/test_cli.py
index 715901e..39bb201 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -183,7 +183,7 @@ class TestOutputFormatter:
     def test_get_severity_style(self):
         """Test severity style mapping."""
         from src.cli.output import OutputFormatter
-        from src.cli.options import ScanOptions
+        from src.cli.options import ScanOptions, SeverityLevel
 
         options = ScanOptions()
         formatter = OutputFormatter(options)