diff --git a/gym-unity/setup.py b/gym-unity/setup.py
index 10069f7f23..7574be3719 100755
--- a/gym-unity/setup.py
+++ b/gym-unity/setup.py
@@ -1,14 +1,14 @@
 #!/usr/bin/env python
 
-from setuptools import setup, Command, find_packages
+from setuptools import setup, find_packages
 
 setup(name='gym_unity',
-      version='0.1.0',
+      version='0.1.1',
       description='Unity Machine Learning Agents Gym Interface',
       license='Apache License 2.0',
       author='Unity Technologies',
       author_email='ML-Agents@unity3d.com',
       url='https://github.com/Unity-Technologies/ml-agents',
       packages=find_packages(),
-      install_requires = ['gym', 'mlagents']
-     )
+      install_requires=['gym', 'mlagents']
+      )