diff --git a/.github/workflows/publish-to-test-pypi.yml b/.github/workflows/publish-to-test-pypi.yml
index 2db691a..e739062 100644
--- a/.github/workflows/publish-to-test-pypi.yml
+++ b/.github/workflows/publish-to-test-pypi.yml
@@ -1,17 +1,16 @@
-name: Publish Python 🐍 distributions 📦 to PyPI and TestPyPI
+name: Build and publish Python 🐍 distributions 📦 to PyPI and TestPyPI
 
 on:
+  workflow_dispatch:
   release:
     types: [published]
 
 jobs:
-  build-n-publish:
-    name: Build and publish Python 🐍 distributions 📦 to PyPI and TestPyPI
+  build:
+    name: Build distribution 📦
     runs-on: ubuntu-22.04
-    permissions:
-      id-token: write
     steps:
-    - uses: actions/checkout@master
+    - uses: actions/checkout@v4
     - name: Set up Python 3.9
       uses: actions/setup-python@v5
       with:
@@ -27,10 +26,40 @@ jobs:
         python -m
         build
         --sdist
-    - name: Publish distribution 📦 to Test PyPI
-      uses: pypa/gh-action-pypi-publish@v1.12.2
+    - name: Store the distribution package
+      uses: actions/upload-artifact@v4
       with:
-        repository-url: https://test.pypi.org/legacy/
-    - name: Publish distribution 📦 to PyPI
-      if: startsWith(github.ref, 'refs/tags')
-      uses: pypa/gh-action-pypi-publish@v1.12.2
+        name: python-package-sdist
+        path: dist/
+
+  publish-to-testpypi:
+    name: Publish Python 🐍 distributions 📦 to TestPyPI
+    runs-on: ubuntu-22.04
+    permissions:
+      id-token: write
+    steps:
+      - name: Download the dist
+        uses: actions/download-artifact@v4
+        with:
+          name: python-package-sdist
+          path: dist/
+      - name: Publish distribution 📦 to Test PyPI
+        uses: pypa/gh-action-pypi-publish@v1.12.2
+        with:
+          repository-url: https://test.pypi.org/legacy/
+
+
+  publish-to-pypi:
+    name: Publish Python 🐍 distributions 📦 to PyPI
+    runs-on: ubuntu-22.04
+    if: startsWith(github.ref, 'refs/tags')
+    permissions:
+      id-token: write
+    steps:
+      - name: Download the dist
+        uses: actions/download-artifact@v4
+        with:
+          name: python-package-sdist
+          path: dist/
+      - name: Publish distribution 📦 to PyPI
+        uses: pypa/gh-action-pypi-publish@v1.12.2
diff --git a/src/dpu_kmeans/_core.pyi b/src/dpu_kmeans/_core.pyi
index 460b38a..659fead 100644
--- a/src/dpu_kmeans/_core.pyi
+++ b/src/dpu_kmeans/_core.pyi
@@ -53,4 +53,4 @@ class Container:
     def pim_cpu_time(self) -> float:
         ...
 FEATURE_TYPE: int = 16
-__version__: str = '0.2.1'
+__version__: str = '0.2.2'