diff --git a/pyproject.toml b/pyproject.toml
index b276d06..b62500f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "yunchang"
-version = "0.5.0"
+version = "0.5.1"
 authors = [
   { name="Jiarui Fang", email="fangjiarui123@gmail.com" },
 ]
diff --git a/yunchang/__init__.py b/yunchang/__init__.py
index 696dee7..ca4ca44 100644
--- a/yunchang/__init__.py
+++ b/yunchang/__init__.py
@@ -4,5 +4,5 @@
 from .globals import set_seq_parallel_pg
 from .comm.extract_local import stripe_extract_local, basic_extract_local, zigzag_extract_local, EXTRACT_FUNC_DICT
 
-__version__ = "0.5.0"
+__version__ = "0.5.1"