diff --git a/stdlib/2/unittest.pyi b/stdlib/2/unittest.pyi
index 43f08a45b..a5cd129b0 100644
--- a/stdlib/2/unittest.pyi
+++ b/stdlib/2/unittest.pyi
@@ -226,7 +226,7 @@ def skip(reason: Union[str, unicode]) -> Any: ...
 class TestProgram:
     result = ...  # type: TestResult
 
-def main(module: str = ..., defaultTest: Optional[str] = ...,
+def main(module: Union[None, Text, types.ModuleType] = ..., defaultTest: Optional[str] = ...,
          argv: Optional[Sequence[str]] = ...,
          testRunner: Union[Type[TextTestRunner], TextTestRunner, None] = ...,
          testLoader: TestLoader = ..., exit: bool = ..., verbosity: int = ...,
diff --git a/stdlib/3/unittest/__init__.pyi b/stdlib/3/unittest/__init__.pyi
index 5b7b37352..3337d00ba 100644
--- a/stdlib/3/unittest/__init__.pyi
+++ b/stdlib/3/unittest/__init__.pyi
@@ -307,7 +307,7 @@ class TextTestRunner(TestRunner):
 class TestProgram:
     result = ...  # type: TestResult
 
-def main(module: str = ...,
+def main(module: Union[None, str, ModuleType] = ...,
          defaultTest: Union[str, Iterable[str], None] = ...,
          argv: Optional[List[str]] = ...,
          testRunner: Union[Type[TestRunner], TestRunner, None] = ...,