diff --git a/pyproject.toml b/pyproject.toml
index 0cef9e1..c852126 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "lightning_trainable"
-version = "0.2.0"
+version = "0.3.1"
 description = "A light-weight lightning_trainable module for pytorch-lightning."
 readme = "README.md"
 requires-python = ">=3.10"