diff --git a/setup.py b/setup.py
index 0e1afc47d06e61448622f32015eb8545fe8a0423..28d5d992a8e73530dd47f9bbd4853490680f2c28 100644
--- a/setup.py
+++ b/setup.py
@@ -1,12 +1,13 @@
+import distutils
+import io
 import os
 import sys
-import io
-from setuptools import setup, find_packages
-import distutils
-from distutils.extension import Extension
 from contextlib import redirect_stdout
+from distutils.extension import Extension
 from importlib import import_module
 
+from setuptools import find_packages, setup
+
 if '--use-cython' in sys.argv:
     USE_CYTHON = True
     sys.argv.remove('--use-cython')
@@ -52,6 +53,7 @@ class SimpleTestRunner(distutils.cmd.Command):
         for test in quick_tests:
             self._run_tests_in_module(test)
 
+
 def readme():
     with open('README.md') as f:
         return f.read()
@@ -69,7 +71,7 @@ def cython_extensions(*extensions):
 try:
     sys.path.insert(0, os.path.abspath('doc'))
     from version_from_git import version_number_from_git
-    version=version_number_from_git()
+    version = version_number_from_git()
     with open("RELEASE-VERSION", "w") as f:
         f.write(version)
 except ImportError:
@@ -88,7 +90,7 @@ setup(name='pystencils',
       packages=['pystencils'] + ['pystencils.' + s for s in find_packages('pystencils')],
       install_requires=['sympy>=1.1', 'numpy', 'appdirs', 'joblib'],
       package_data={'pystencils': ['include/*.h', 'backends/cuda_known_functions.txt']},
-      ext_modules = cython_extensions("pystencils.boundaries.createindexlistcython"),
+      ext_modules=cython_extensions("pystencils.boundaries.createindexlistcython"),
       classifiers=[
           'Development Status :: 4 - Beta',
           'Framework :: Jupyter',