diff --git a/setup.py b/setup.py
index 743da7385..ed12dfe0f 100644
--- a/setup.py
+++ b/setup.py
@@ -8,7 +8,10 @@
 import sys
 from pathlib import Path
 import setuptools
-from wheel.bdist_wheel import bdist_wheel
+try:
+    from setuptools.command.bdist_wheel import bdist_wheel
+except ImportError:
+    from wheel.bdist_wheel import bdist_wheel
 from setuptools.command.build_py import build_py as build_py_orig
 
 sys.path.insert(0, str(Path(__file__).parent / "setupsrc"))