--- a/build_scripts/wheel_override.py
+++ b/build_scripts/wheel_override.py
@@ -20,9 +20,9 @@
 
     from packaging import tags
     from wheel import __version__ as wheel_version
-    from wheel.bdist_wheel import bdist_wheel as _bdist_wheel
-    from wheel.bdist_wheel import get_abi_tag, get_platform
-    from wheel.bdist_wheel import safer_name as _safer_name
+    from setuptools.command.bdist_wheel import bdist_wheel as _bdist_wheel
+    from setuptools.command.bdist_wheel import get_abi_tag, get_platform
+    from setuptools.command.bdist_wheel import safer_name as _safer_name
 
     wheel_module_exists = True
 except Exception as e: