diff --git a/mctx/__init__.py b/mctx/__init__.py
index 13daedc..26a409d 100644
--- a/mctx/__init__.py
+++ b/mctx/__init__.py
@@ -37,7 +37,7 @@
 from mctx._src.search import search
 from mctx._src.tree import Tree
 
-__version__ = "0.0.3"
+__version__ = "0.0.4"
 
 __all__ = (
     "ChanceRecurrentFnOutput",