diff --git a/setup.py b/setup.py
index 5da19bb..e69132c 100644
--- a/setup.py
+++ b/setup.py
@@ -43,7 +43,7 @@
 
 setuptools.setup(
     name="torch-lr-finder",
-    version="0.2.1",
+    version="0.2.2",
     author="David Silva",
     author_email="davidtvs10@gmail.com",
     description="Pytorch implementation of the learning rate range test",