diff --git a/pyproject.toml b/pyproject.toml
index 78dfc6a..6853329 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "tkan" 
-version = "0.4.0" 
+version = "0.4.1" 
 description = "Temporal KAN model"
 authors = [ "Rémi Genet", "Hugo Inzirillo"]  
 readme = "README.md"