diff --git a/pyproject.toml b/pyproject.toml
index de91dbf..8acbf93 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -67,7 +67,7 @@ version = {attr = "keep_gpu.__version__"}
 # -----------------------------
 # bump-my-version configuration
 [tool.bumpversion]
-current_version = "0.3.2"
+current_version = "0.4.0"
 commit          = true
 tag             = true
 tag_name        = "v{new_version}"
diff --git a/src/keep_gpu/__init__.py b/src/keep_gpu/__init__.py
index b68bc92..61bb0a7 100644
--- a/src/keep_gpu/__init__.py
+++ b/src/keep_gpu/__init__.py
@@ -2,4 +2,4 @@
 
 __author__ = """Siyuan Wang"""
 __email__ = "sywang0227@gmail.com"
-__version__ = "0.3.2"
+__version__ = "0.4.0"