diff --git a/stdlib/2/unittest.pyi b/stdlib/2/unittest.pyi
index f79ec5e86843..8e8665034a7f 100644
--- a/stdlib/2/unittest.pyi
+++ b/stdlib/2/unittest.pyi
@@ -192,12 +192,12 @@ class TestLoader:
     suiteClass = ...  # type: Callable[[List[TestCase]], TestSuite]
     def loadTestsFromTestCase(self,
                               testCaseClass: Type[TestCase]) -> TestSuite: ...
-    def loadTestsFromModule(self, module: str = ...,
+    def loadTestsFromModule(self, module: types.ModuleType = ...,
                             use_load_tests: bool = ...) -> TestSuite: ...
     def loadTestsFromName(self, name: str = ...,
-                          module: Optional[str] = ...) -> TestSuite: ...
+                          module: Optional[types.ModuleType] = ...) -> TestSuite: ...
     def loadTestsFromNames(self, names: List[str] = ...,
-                          module: Optional[str] = ...) -> TestSuite: ...
+                          module: Optional[types.ModuleType] = ...) -> TestSuite: ...
     def discover(self, start_dir: str, pattern: str = ...,
                  top_level_dir: Optional[str] = ...) -> TestSuite: ...
     def getTestCaseNames(self, testCaseClass: Type[TestCase] = ...) -> List[str]: ...