diff --git a/pyproject.toml b/pyproject.toml
index 78ed7e3..5a405d0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "git-t5"
-version = "0.2.1"
+version = "0.2.2"
 description = "Open source machine learning framework for training T5 models on source code in JAX/Flax."
 authors = ["mozharovsky <mozharovsky@live.com>"]
 maintainers = ["mozharovsky <mozharovsky@live.com>"]