diff --git a/mypy/stubgen.py b/mypy/stubgen.py
index 2d3e8e8f48ef..34d01b337bac 100755
--- a/mypy/stubgen.py
+++ b/mypy/stubgen.py
@@ -1704,14 +1704,14 @@ def parse_options(args: List[str]) -> Options:
                    export_less=ns.export_less)
 
 
-def main() -> None:
+def main(args: Optional[List[str]] = None) -> None:
     mypy.util.check_python_version('stubgen')
     # Make sure that the current directory is in sys.path so that
     # stubgen can be run on packages in the current directory.
     if not ('' in sys.path or '.' in sys.path):
         sys.path.insert(0, '')
 
-    options = parse_options(sys.argv[1:])
+    options = parse_options(sys.argv[1:] if args is None else args)
     generate_stubs(options)