diff --git a/.github/workflows/check-release.yml b/.github/workflows/check-release.yml
index c52bceb99..b0dd084ad 100644
--- a/.github/workflows/check-release.yml
+++ b/.github/workflows/check-release.yml
@@ -69,9 +69,9 @@ jobs:
           hatch build
       - name: Set PyPI token
         run: |
-          if [ "${{ needs.check-version.outputs.PACKAGE_NAME }}" == "vizro-core" ]; then
+          if [ "${{ needs.check-version.outputs.package_name }}" == "vizro-core" ]; then
             echo 'PYPI_TOKEN=${{ secrets.VIZRO_PYPI_TOKEN }}' >> $GITHUB_ENV
-          elif [ "${{ needs.check-version.outputs.PACKAGE_NAME }}" == "vizro-ai" ]; then
+          elif [ "${{ needs.check-version.outputs.package_name }}" == "vizro-ai" ]; then
             echo 'PYPI_TOKEN=${{ secrets.VIZRO_AI_PYPI_TOKEN }}' >> $GITHUB_ENV
           fi
       - name: Publish distribution 📦 to PyPI
@@ -82,10 +82,10 @@ jobs:
       - name: Check correct package uploaded to PyPI
         run: |
           cd "${{ needs.check-version.outputs.package_name }}/dist"
-          local=$(md5sum vizro-${{needs.check-version.outputs.package_version}}-py3-none-any.whl)
+          local=$(md5sum ${{ needs.check-version.outputs.package_name }}-${{needs.check-version.outputs.package_version}}-py3-none-any.whl)
           cd ..
-          pip download vizro==${{needs.check-version.outputs.package_version}} -d . --no-deps --timeout 300
-          pypi=$(md5sum vizro-${{needs.check-version.outputs.package_version}}-py3-none-any.whl)
+          pip download ${{ needs.check-version.outputs.package_name }}==${{needs.check-version.outputs.package_version}} -d . --no-deps --timeout 300
+          pypi=$(md5sum ${{ needs.check-version.outputs.package_name }}-${{needs.check-version.outputs.package_version}}-py3-none-any.whl)
           if [[ $local = $pypi ]]; then echo "md5 hash is the same"; else echo "md5 hash is not the same"; exit 1; fi
 
   version-bump: