diff --git a/.github/workflows/gen_whl_to_pypi_rapid_layout.yml b/.github/workflows/publish_whl.yml
similarity index 93%
rename from .github/workflows/gen_whl_to_pypi_rapid_layout.yml
rename to .github/workflows/publish_whl.yml
index 0fc50c5..2e2fae8 100644
--- a/.github/workflows/gen_whl_to_pypi_rapid_layout.yml
+++ b/.github/workflows/publish_whl.yml
@@ -34,7 +34,7 @@ jobs:
           cp $DIR_NAME/*.onnx rapid_layout/models/
           pip install -r requirements.txt
           pip install pytest wheel get_pypi_latest_version
-          pytest tests/test_layout.py
+          pytest tests/test*.py
 
   GenerateWHL_PushPyPi:
     needs: UnitTesting
@@ -51,7 +51,7 @@ jobs:
           DIR_NAME=${ZIP_NAME%.*}
           unzip $ZIP_NAME
           mv $DIR_NAME/*.onnx rapid_layout/models/
-          python setup_layout.py bdist_wheel ${{ github.ref_name }}
+          python setup.py bdist_wheel ${{ github.ref_name }}
 
       - name: Publish distribution 📦 to PyPI
         uses: pypa/gh-action-pypi-publish@v1.5.0