diff --git a/pymc_bart/__init__.py b/pymc_bart/__init__.py
index 452b6ce..613f040 100644
--- a/pymc_bart/__init__.py
+++ b/pymc_bart/__init__.py
@@ -25,7 +25,7 @@
 )
 
 __all__ = ["BART", "PGBART"]
-__version__ = "0.5.3"
+__version__ = "0.5.4"
 
 
 pm.STEP_METHODS = list(pm.STEP_METHODS) + [PGBART]