diff --git a/beobench/__init__.py b/beobench/__init__.py
index 40d66695..0d7427e6 100644
--- a/beobench/__init__.py
+++ b/beobench/__init__.py
@@ -2,6 +2,6 @@
 
 __author__ = """rdnfn"""
 __email__ = "-"
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 
 from beobench.utils import restart
diff --git a/setup.cfg b/setup.cfg
index a2566cf7..872b1dde 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.2.0
+current_version = 0.2.1
 commit = True
 tag = True
 
diff --git a/setup.py b/setup.py
index 06e012c7..7f76b474 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@
 with open("HISTORY.rst", encoding="UTF-8") as history_file:
     history = history_file.read()
 
-version = "0.2.0"
+version = "0.2.1"
 
 requirements = [
     "wandb",