diff --git a/setup.py b/setup.py
index 7bca59a6..a8a5b440 100644
--- a/setup.py
+++ b/setup.py
@@ -7,7 +7,7 @@
     
 setuptools.setup(
     name='pytorch-lifestream',
-    version='0.5.1',
+    version='0.5.2',
     author='',
     author_email='',
     description='Lifestream data analysis with PyTorch',