diff --git a/.github/workflows/python-ci.yml b/.github/workflows/python-ci.yml
index 9a82135..9ec61f6 100644
--- a/.github/workflows/python-ci.yml
+++ b/.github/workflows/python-ci.yml
@@ -54,7 +54,7 @@ jobs:
         echo "coverage=${COVERAGE_PERCENT}" >> $GITHUB_ENV
 
     - name: Create mypy warning badge
-      if: ${{ steps.branch_name.outputs.branch_name == 'main' }}
+      if: ${{ github.ref_name == 'main' }}
       uses: schneegans/dynamic-badges-action@v1.7.0
       with:
         auth: ${{ secrets.GIST_SECRET }}
@@ -68,7 +68,7 @@ jobs:
         invertColorRange: true
 
     - name: Create pylint badge
-      if: ${{ steps.branch_name.outputs.branch_name == 'main' }}
+      if: ${{ github.ref_name == 'main' }}
       uses: schneegans/dynamic-badges-action@v1.7.0
       with:
         auth: ${{ secrets.GIST_SECRET }}
@@ -81,7 +81,7 @@ jobs:
         maxColorRange: 9
 
     - name: Create coverage badge
-      if: ${{ steps.branch_name.outputs.branch_name == 'main' }}
+      if: ${{ github.ref_name == 'main' }}
       uses: schneegans/dynamic-badges-action@v1.7.0
       with:
         auth: ${{ secrets.GIST_SECRET }}