diff --git a/stdlib/2/unittest.pyi b/stdlib/2/unittest.pyi
index a2a6af215..ce06d3539 100644
--- a/stdlib/2/unittest.pyi
+++ b/stdlib/2/unittest.pyi
@@ -6,7 +6,7 @@
 
 from typing import (
     Any, Callable, Dict, Iterable, Tuple, List, TextIO, Sequence,
-    overload, TypeVar, Pattern
+    overload, Set, TypeVar, Pattern
 )
 from abc import abstractmethod, ABCMeta
 
diff --git a/stdlib/3/unittest/__init__.pyi b/stdlib/3/unittest/__init__.pyi
index 14b2f3e24..0e074099f 100644
--- a/stdlib/3/unittest/__init__.pyi
+++ b/stdlib/3/unittest/__init__.pyi
@@ -2,7 +2,7 @@
 
 from typing import (
     Any, Callable, Dict, Iterable, Iterator, List, Optional, Pattern, Sequence,
-    TextIO, Tuple, Type, TypeVar, Union,
+    Set, TextIO, Tuple, Type, TypeVar, Union,
     overload,
 )
 import logging