diff --git a/setup.py b/setup.py
index 8a6a969..e47f2fa 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="visualkeras",
-    version="0.0.1",
+    version="0.0.2",
     author="Paul Gavrikov",
     author_email="paul.gavrikov@hs-offenburg.de",
     description="Architecture visualization of Keras models",