diff --git a/pyproject.toml b/pyproject.toml
index d4cbc5e6..3428f333 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "openpipe-art"
-version = "0.5.1"
+version = "0.5.2"
 description = "The OpenPipe Agent Reinforcement Training (ART) library"
 readme = "README.md"
 requires-python = ">=3.10"
diff --git a/uv.lock b/uv.lock
index 3794f09f..d2ba5a80 100644
--- a/uv.lock
+++ b/uv.lock
@@ -4121,7 +4121,7 @@ wheels = [
 
 [[package]]
 name = "openpipe-art"
-version = "0.5.1"
+version = "0.5.2"
 source = { editable = "." }
 dependencies = [
     { name = "litellm" },