diff --git a/src/litmodels/__about__.py b/src/litmodels/__about__.py
index 773e9ee..b801271 100644
--- a/src/litmodels/__about__.py
+++ b/src/litmodels/__about__.py
@@ -1,4 +1,4 @@
-__version__ = "0.0.1.dev"
+__version__ = "0.0.2.dev"
 __author__ = "Lightning-AI et al."
 __author_email__ = "community@lightning.ai"
 __license__ = "Apache-2.0"  # TODO: change if needed and update the LICENSE file