diff --git a/elegy/__init__.py b/elegy/__init__.py
index 65dfb9dc..264d0a7c 100644
--- a/elegy/__init__.py
+++ b/elegy/__init__.py
@@ -1,6 +1,6 @@
 # isort:skip_file
 
-__version__ = "0.8.3"
+__version__ = "0.8.4"
 
 from treex import *
 
diff --git a/pyproject.toml b/pyproject.toml
index 669e3b01..93d2daeb 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.3"
+version = "0.8.4"
 license = "APACHE"
 readme = "README.md"
 repository = "https://github.com/poets-ai/elegy"