diff --git a/setup.py b/setup.py
index f133f78..5147e7a 100644
--- a/setup.py
+++ b/setup.py
@@ -19,7 +19,7 @@
 
 setup(
     name='tf_unet',
-    version='0.1.1',
+    version='0.1.2',
     description='Unet TensorFlow implementation',
     long_description=readme + '\n\n' + history,
     author='Joel Akeret',
diff --git a/tf_unet/__init__.py b/tf_unet/__init__.py
index 84b5898..7d6bb1c 100644
--- a/tf_unet/__init__.py
+++ b/tf_unet/__init__.py
@@ -1,3 +1,3 @@
 __author__ = 'Joel Akeret'
-__version__ = '0.1.1'
+__version__ = '0.1.2'
 __credits__ = 'ETH Zurich, Institute for Astronomy'