diff --git a/elegy/__init__.py b/elegy/__init__.py
index 232b8c6f..a61afa35 100644
--- a/elegy/__init__.py
+++ b/elegy/__init__.py
@@ -1,6 +1,6 @@
 # isort:skip_file
 
-__version__ = "0.8.1"
+__version__ = "0.8.2"
 
 from treex import *
 
diff --git a/pyproject.toml b/pyproject.toml
index 586750f5..1fe4f0d1 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.1"
+version = "0.8.2"
 license = "APACHE"
 readme = "README.md"
 repository = "https://github.com/poets-ai/elegy"