diff --git a/jax_metrics/__init__.py b/jax_metrics/__init__.py
index 3cfabe8..dc20118 100644
--- a/jax_metrics/__init__.py
+++ b/jax_metrics/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.1.1"
+__version__ = "0.1.2"
 
 
 from jax_metrics.losses import Loss