diff --git a/python/fedml/__init__.py b/python/fedml/__init__.py
index 0b4763e674..3e0929d56e 100644
--- a/python/fedml/__init__.py
+++ b/python/fedml/__init__.py
@@ -23,7 +23,7 @@
 _global_training_type = None
 _global_comm_backend = None
 
-__version__ = "0.7.316"
+__version__ = "0.7.317"
 
 
 def init(args=None):
diff --git a/python/setup.py b/python/setup.py
index 2910d1cc94..cf28885c37 100755
--- a/python/setup.py
+++ b/python/setup.py
@@ -5,9 +5,8 @@
 from setuptools import setup, find_packages
 
 
-from wheel.bdist_wheel import bdist_wheel
-#
-# try:
+try:
+     from wheel.bdist_wheel import bdist_wheel
 #     from wheel.bdist_wheel import bdist_wheel as _bdist_wheel
 #
 #
@@ -19,8 +18,8 @@
 #
 #
 #
-# except ImportError:
-#     bdist_wheel = None
+except ImportError:
+    bdist_wheel = None
 
 requirements = [
     "numpy",
@@ -75,7 +74,7 @@
 
 setup(
     name="fedml",
-    version="0.7.316",
+    version="0.7.317",
     author="FedML Team",
     author_email="ch@fedml.ai",
     description="A research and production integrated edge-cloud library for "