diff --git a/declearn/__init__.py b/declearn/__init__.py
index 63f487493e2c3eaf77067659ac3e4b77cdb90f59..e35504e03adf0b9a7edb07e034a50b2cc8e42a7a 100644
--- a/declearn/__init__.py
+++ b/declearn/__init__.py
@@ -64,4 +64,4 @@ from . import (
     utils,
 )
 
-__version__ = "2.0.1"
+__version__ = "2.0.2"
diff --git a/pyproject.toml b/pyproject.toml
index 67e75153a0d87d40c1911f687bafd6c5508ab126..be3bc4cb5ad2a1885477d8f1014274e2bd5a23fe 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -8,7 +8,7 @@ requires = [
 
 [project]
 name = "declearn"
-version = "2.0.1"
+version = "2.0.2"
 description = "Declearn - a python package for private decentralized learning."
 readme = "README.md"
 requires-python = ">=3.8"