diff --git a/shell_alias_gen/cli.py b/shell_alias_gen/cli.py
index 63dc66e..5b6d73e 100644
--- a/shell_alias_gen/cli.py
+++ b/shell_alias_gen/cli.py
@@ -238,7 +238,7 @@ def locations():
             console.print()
 
 
-def _get_parser(shell: str, history_file: Optional[Path]) -> Optional["BaseParser"]:
+def _get_parser(shell: str, history_file: Optional[Path]) -> Optional[HistoryParser]:
     """Get appropriate history parser."""
     if shell == 'auto' and history_file:
         detected = HistoryParserFactory.detect_shell_from_file(str(history_file))