diff --git a/setup.py b/setup.py
index b9cde6a..11003a2 100755
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@
 
 setup(
     name='torch_stoi',
-    version='0.1.2',
+    version='0.2.0',
     description='Computes Short Term Objective Intelligibility in PyTorch',
     author='Manuel Pariente',
     author_email='pariente.mnl@gmail.com',
diff --git a/torch_stoi/__init__.py b/torch_stoi/__init__.py
index 9006c41..98c4faa 100755
--- a/torch_stoi/__init__.py
+++ b/torch_stoi/__init__.py
@@ -1,4 +1,4 @@
 from .stoi import NegSTOILoss
 
 
-__version__ = '0.1.2'
+__version__ = '0.2.0'