diff --git a/shell_alias_gen/cli.py b/shell_alias_gen/cli.py
index 1501e70..63dc66e 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]):
+def _get_parser(shell: str, history_file: Optional[Path]) -> Optional["BaseParser"]:
     """Get appropriate history parser."""
     if shell == 'auto' and history_file:
         detected = HistoryParserFactory.detect_shell_from_file(str(history_file))
@@ -251,7 +251,7 @@ def _get_parser(shell: str, history_file: Optional[Path]):
     return HistoryParserFactory.get_parser(shell)
 
 
-def _get_rcfile_path(shell: str):
+def _get_rcfile_path(shell: str) -> Optional[Path]:
     """Get the rc file path for a shell."""
     from os.path import expanduser