diff --git a/elegy/__init__.py b/elegy/__init__.py
index 47c02941..5b11edb1 100644
--- a/elegy/__init__.py
+++ b/elegy/__init__.py
@@ -1,6 +1,6 @@
 # isort:skip_file
 
-__version__ = "0.8.5"
+__version__ = "0.8.6"
 
 from treex import *
 
diff --git a/pyproject.toml b/pyproject.toml
index 52b9acbd..8231770b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -5,7 +5,7 @@ authors = ["Cristian Garcia <cgarcia.e88@gmail.com>",
            "Carlos Alvarez <candres.alv@gmail.com>",
            "David Cardozo <davidcardozo@berkeley.edu>",
            "Sebastian Arango"]
-version = "0.8.5"
+version = "0.8.6"
 license = "APACHE"
 readme = "README.md"
 repository = "https://github.com/poets-ai/elegy"