diff --git a/jax_toolkit/_version.py b/jax_toolkit/_version.py
index 485f44a..b3f4756 100644
--- a/jax_toolkit/_version.py
+++ b/jax_toolkit/_version.py
@@ -1 +1 @@
-__version__ = "0.1.1"
+__version__ = "0.1.2"
diff --git a/setup.cfg b/setup.cfg
index b1c0d9e..4dd580b 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.1.1
+current_version = 0.1.2
 commit = True
 tag = True