diff --git a/node2vec/__init__.py b/node2vec/__init__.py
index 21cca28..3deeb9a 100644
--- a/node2vec/__init__.py
+++ b/node2vec/__init__.py
@@ -1,4 +1,4 @@
 from .node2vec import Node2Vec
 from . import edges
 
-__version__ = '0.2.0'
\ No newline at end of file
+__version__ = '0.2.1'
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 02920ee..2555765 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@
 setup(
     name='node2vec',
     packages=['node2vec'],
-    version='0.2.0',
+    version='0.2.1',
     description='Implementation of the node2vec algorithm.',
     author='Elior Cohen',
     author_email='elior.cohen.p@gmail.com',