diff --git a/setup.py b/setup.py
index ebf065e..8bcec1f 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="gradient-accumulator",
-    version="0.2.1",
+    version="0.2.2",
     author="André Pedersen and David Bouget",
     author_email="andrped94@gmail.com",
     description="Package for gradient accumulation in TensorFlow",