Update setup.py

setup.py

@@ -2,7 +2,7 @@
 from setuptools import setup, find_packages
 from setuptools.command.develop import develop
 from setuptools.command.install import install
-from setuptools.command.bdist_wheel import bdist_wheel
+from setuptools.command.bdist import bdist
 import cffi
 import tomli
 from typing import Dict, Any
@@ -45,7 +45,7 @@ def build_cffi():
 
     ffibuilder.compile(verbose=True)
 
-class TMUbdistWheel(bdist_wheel):
+class TMUbdistWheel(bdist):
     """
     Custom install command that builds the CFFI modules
     before proceeding with the standard installation process.
@@ -89,7 +89,7 @@ def has_ext_modules(foo):
     cmdclass={
         "install": TMUInstall,
         "develop": TMUDevelop,
-        "bdist_wheel": TMUbdistWheel
+        "bdist": TMUbdistWheel
     },
     distclass=BinaryDistribution
 )