diff --git a/.github/workflows/package-publish.yml b/.github/workflows/package-publish.yml
index c1d8f3b..512bc40 100644
--- a/.github/workflows/package-publish.yml
+++ b/.github/workflows/package-publish.yml
@@ -34,11 +34,11 @@ jobs:
         user: __token__
         password: ${{ secrets.TEST_PYPI_TOKEN }}
         repository_url: https://test.pypi.org/legacy/
-      if: ${{ github.event.inputs.pypi_repo == 'testpypi' }}
+      if: ${{ github.event.inputs.pypi_repo != 'pypi' }}
 
     - name: Publish package to PyPI
       uses: pypa/gh-action-pypi-publish@release/v1
       with:
         user: __token__
         password: ${{ secrets.PYPI_TOKEN }}
-      if: ${{ github.event.inputs.pypi_repo == 'pypi' }}
\ No newline at end of file
+      if: ${{ github.event.inputs.pypi_repo == 'pypi' || github.event_name == 'release' }}
\ No newline at end of file