diff --git a/setup.py b/setup.py
index ceddf40..10eac0e 100644
--- a/setup.py
+++ b/setup.py
@@ -9,9 +9,13 @@
 import subprocess
 from warnings import warn
 import packaging.version
-from wheel.bdist_wheel import bdist_wheel
 import sysconfig
 
+try:
+    from setuptools.command.bdist_wheel import bdist_wheel
+except ImportError:
+    from wheel.bdist_wheel import bdist_wheel
+
 
 LIBERFADIR = os.path.join('liberfa', 'erfa')
 ERFA_SRC = os.path.join(LIBERFADIR, 'src')