diff --git a/tf2jax/__init__.py b/tf2jax/__init__.py
index 21b05d6..db1771a 100644
--- a/tf2jax/__init__.py
+++ b/tf2jax/__init__.py
@@ -25,7 +25,7 @@
 from tf2jax._src.tf2jax import convert_functional
 from tf2jax._src.tf2jax import convert_functional_from_restored
 
-__version__ = "0.3.0"
+__version__ = "0.3.1"
 
 #  _________________________________________
 # / Please don't use symbols in `_src` they \