diff --git a/.github/workflows/table_cls.yml b/.github/workflows/table_cls.yml
index eb9768b..ee229b3 100644
--- a/.github/workflows/table_cls.yml
+++ b/.github/workflows/table_cls.yml
@@ -3,10 +3,10 @@ name: Push table_cls to pypi
 on:
   push:
     branches: [ main ]
-    paths:
-      - 'table_cls/**'
-    # tags:
-    #   - v*
+    # paths:
+    #   - 'table_cls/**'
+    tags:
+      - table_cls_v*
 
 jobs:
   UnitTesting:
@@ -35,33 +35,33 @@ jobs:
 
           pytest tests/test_table_cls.py
 
-#  GenerateWHL_PushPyPi:
-#    needs: UnitTesting
-#    runs-on: ubuntu-latest
-#
-#    steps:
-#      - uses: actions/checkout@v3
-#
-#      - name: Set up Python 3.10
-#        uses: actions/setup-python@v4
-#        with:
-#          python-version: '3.10'
-#          architecture: 'x64'
-#
-#      - name: Run setup.py
-#        run: |
-#          pip install -r requirements.txt
-#          python -m pip install --upgrade pip
-#          pip install wheel get_pypi_latest_version
-#
-#          wget https://github.com/RapidAI/TableStructureRec/releases/download/v0.0.0/table_cls_models.zip
-#          unzip table_cls_models.zip
-#          mv table_cls_models/*.onnx table_cls/models/
-#
-#          python setup_table_cls.py bdist_wheel "${{ github.event.head_commit.message }}"
-#
-#      - name: Publish distribution 📦 to PyPI
-#        uses: pypa/gh-action-pypi-publish@v1.5.0
-#        with:
-#          password: ${{ secrets.TABLE_CLS }}
-#          packages_dir: dist/
+  GenerateWHL_PushPyPi:
+    needs: UnitTesting
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v3
+
+      - name: Set up Python 3.10
+        uses: actions/setup-python@v4
+        with:
+          python-version: '3.10'
+          architecture: 'x64'
+
+      - name: Run setup.py
+        run: |
+          pip install -r requirements.txt
+          python -m pip install --upgrade pip
+          pip install wheel get_pypi_latest_version
+
+          wget https://github.com/RapidAI/TableStructureRec/releases/download/v0.0.0/table_cls_models.zip
+          unzip table_cls_models.zip
+          mv table_cls_models/*.onnx table_cls/models/
+
+          python setup_table_cls.py bdist_wheel "${{ github.ref_name }}"
+
+      - name: Publish distribution 📦 to PyPI
+        uses: pypa/gh-action-pypi-publish@v1.5.0
+        with:
+          password: ${{ secrets.TABLE_CLS }}
+          packages_dir: dist/
diff --git a/.github/workflows/wired_table_rec.yml b/.github/workflows/wired_table_rec.yml
index 2dae757..0f3c3dd 100644
--- a/.github/workflows/wired_table_rec.yml
+++ b/.github/workflows/wired_table_rec.yml
@@ -3,10 +3,10 @@ name: Push wired_table_rec to pypi
 on:
   push:
     branches: [ main ]
-    paths:
-      - 'wired_table_rec/**'
-    # tags:
-    #   - v*
+    # paths:
+    #   - 'wired_table_rec/**'
+    tags:
+      - wired_v*
 
 jobs:
   UnitTesting:
@@ -62,7 +62,7 @@ jobs:
          unzip wired_table_rec_models.zip
          mv wired_table_rec_models/*.onnx wired_table_rec/models/
 
-         python setup_wired.py bdist_wheel "${{ github.event.head_commit.message }}"
+         python setup_wired.py bdist_wheel "${{ github.ref_name }}"
 
      - name: Publish distribution 📦 to PyPI
        uses: pypa/gh-action-pypi-publish@v1.5.0