diff --git a/pyproject.toml b/pyproject.toml
index 64586f24..6f23666c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "atroposlib"
-version = "0.3.0"
+version = "0.4.0"
 description = "Atropos: An Environment and Rollout handler for LLM RL"
 readme = "README.md"
 requires-python = ">=3.10"