diff --git a/mctx/__init__.py b/mctx/__init__.py
index 49dc741..4f741b7 100644
--- a/mctx/__init__.py
+++ b/mctx/__init__.py
@@ -33,7 +33,7 @@
 from mctx._src.search import search
 from mctx._src.tree import Tree
 
-__version__ = "0.0.1"
+__version__ = "0.0.2"
 
 __all__ = (
     "GumbelMuZeroExtraData",