diff --git a/Cargo.lock b/Cargo.lock
index fbdf2ea..3f66271 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -313,7 +313,7 @@ dependencies = [
 
 [[package]]
 name = "omikuji"
-version = "0.4.0"
+version = "0.4.1"
 dependencies = [
  "assert_approx_eq",
  "clap",
diff --git a/Cargo.toml b/Cargo.toml
index ad004fe..0019c79 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,7 +1,7 @@
 [package]
 name = "omikuji"
 description = "an efficient implementation of Partitioned Label Treesand its variations for extreme multi-label classification"
-version = "0.4.0"
+version = "0.4.1"
 edition = "2021"
 authors = ["Tom Dong <tom.tung.dyb@gmail.com>"]
 license = "MIT"
diff --git a/c-api/Cargo.lock b/c-api/Cargo.lock
index b2c08a8..3b7ec50 100644
--- a/c-api/Cargo.lock
+++ b/c-api/Cargo.lock
@@ -345,7 +345,7 @@ dependencies = [
 
 [[package]]
 name = "omikuji"
-version = "0.4.0"
+version = "0.4.1"
 dependencies = [
  "hashbrown",
  "itertools",
@@ -366,7 +366,7 @@ dependencies = [
 
 [[package]]
 name = "omikuji_c-api"
-version = "0.4.0"
+version = "0.4.1"
 dependencies = [
  "cbindgen",
  "itertools",
diff --git a/c-api/Cargo.toml b/c-api/Cargo.toml
index db4b5fe..179cf3e 100644
--- a/c-api/Cargo.toml
+++ b/c-api/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "omikuji_c-api"
-version = "0.4.0"
+version = "0.4.1"
 edition = "2021"
 authors = ["Tom Dong <tom.tung.dyb@gmail.com>"]
 
diff --git a/python-wrapper/omikuji/__init__.py b/python-wrapper/omikuji/__init__.py
index e1bd7f4..e7f1f9a 100644
--- a/python-wrapper/omikuji/__init__.py
+++ b/python-wrapper/omikuji/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.4.0"
+__version__ = "0.4.1"
 __all__ = ["Model", "LossType"]
 
 from ._libomikuji import lib, ffi
diff --git a/setup.py b/setup.py
index 50f3d1e..cab3745 100644
--- a/setup.py
+++ b/setup.py
@@ -28,7 +28,7 @@ def load_readme():
 
 setup(
     name="omikuji",
-    version="0.4.0",
+    version="0.4.1",
     author="Tom Dong",
     author_email="tom.tung.dyb@gmail.com",
     description=(