diff --git a/ibis/common/collections.py b/ibis/common/collections.py
index d0e56a85c0f75..c0020d77e5150 100644
--- a/ibis/common/collections.py
+++ b/ibis/common/collections.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from types import MappingProxyType
-from typing import Any, Hashable, Iterable, Iterator, Mapping, TypeVar
+from typing import Any, Hashable, Iterable, Iterator, Mapping, Set, TypeVar
 
 from public import public
 from typing_extensions import Self
@@ -208,7 +208,7 @@ def __repr__(self):
         return f"{self.__class__.__name__}({super().__repr__()})"
 
 
-class DisjointSet(Mapping[K, set[K]]):
+class DisjointSet(Mapping[K, Set[K]]):
     """Disjoint set data structure.
 
     Also known as union-find data structure. It is a data structure that keeps