diff --git a/setup.py b/setup.py
index b0bfdd2..5748256 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@
       description='Distributed Deep learning with Apache Spark with Keras.',
       url='https://github.com/JoeriHermans/dist-keras',
       author='Joeri Hermans',
-      version='0.2.0',
+      version='0.2.1',
       author_email='joeri@joerihermans.com',
       license='GPLv3',
       install_requires=['theano', 'tensorflow', 'keras', 'flask'],