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