diff --git a/setup.py b/setup.py
index cf5756f..d35f291 100644
--- a/setup.py
+++ b/setup.py
@@ -1,8 +1,8 @@
 import setuptools
-from pkg_resources import get_distribution
+from importlib.metadata import version
 
 try:
-    get_distribution("setuptools>=39.2.0")
+    assert tuple(int(x) for x in version("setuptools").split(".")[:2]) >= (39, 2)
 except Exception as e:
     raise AssertionError(
         "Please upgrade setuptools by `pip install -U setuptools`: {}".format(