diff --git a/declearn/__init__.py b/declearn/__init__.py
index 075582a5e0253267a303617eb869ae4cc79aee55..63f487493e2c3eaf77067659ac3e4b77cdb90f59 100644
--- a/declearn/__init__.py
+++ b/declearn/__init__.py
@@ -23,9 +23,9 @@ with the aim of providing users with a modular and extensible framework
 to implement federated learning algorithms and apply them to real-world
 (or simulated) data using any common machine learning framework.
 
-Declearn provides with abstractions that enable algorithms to be written 
-agnostic to the actual computation framework as well as with workable 
-interfaces that cover some of the most popular frameworks, such as 
+Declearn provides with abstractions that enable algorithms to be written
+agnostic to the actual computation framework as well as with workable
+interfaces that cover some of the most popular frameworks, such as
 Scikit-Learn, TensorFlow and PyTorch.
 
 The package is organized into the following submodules:
@@ -64,4 +64,4 @@ from . import (
     utils,
 )
 
-__version__ = "2.0.0"
+__version__ = "2.0.1"
diff --git a/pyproject.toml b/pyproject.toml
index 919316b36113d0e286fa2668f6eb814750f32cbf..e4e15c370f8ba64486abb45dcd60cc4e5f3655ca 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -8,7 +8,7 @@ requires = [
 
 [project]
 name = "declearn"
-version = "2.0.0"
+version = "2.0.1"
 description = "Declearn - a python package for private decentralized learning."
 readme = "README.md"
 requires-python = ">=3.8"