diff --git a/distrax/__init__.py b/distrax/__init__.py
index 96ee760..5ac069a 100644
--- a/distrax/__init__.py
+++ b/distrax/__init__.py
@@ -65,7 +65,7 @@
 from distrax._src.utils.monte_carlo import mc_estimate_mode
 from distrax._src.utils.transformations import register_inverse
 
-__version__ = "0.0.1"
+__version__ = "0.0.2"
 
 __all__ = (
     "as_bijector",