diff --git a/.github/workflows/reusable_build_and_upload_wheels.yml b/.github/workflows/reusable_build_and_upload_wheels.yml
index af255c235557..432944479aa3 100644
--- a/.github/workflows/reusable_build_and_upload_wheels.yml
+++ b/.github/workflows/reusable_build_and_upload_wheels.yml
@@ -227,7 +227,7 @@ jobs:
 
       - name: "Build rerun_notebook"
         # only build the notebook if we are building for pypi and running linux-x64
-        if: ${{ inputs.MODE == 'pypi' && inputs.PLATFORM == 'linux-x64' }}
+        if: ${{ (inputs.MODE == 'pypi' || inputs.MODE == 'extra') && inputs.PLATFORM == 'linux-x64' }}
         run: |
           rm -rf dist
           pixi run js-build-base
@@ -235,7 +235,7 @@ jobs:
             --dir commit/${{ steps.get-sha.outputs.sha }}/wheels
 
       - name: Save rerun_notebook wheel artifact
-        if: ${{ inputs.MODE == 'pypi' && inputs.PLATFORM == 'linux-x64' }}
+        if: ${{ (inputs.MODE == 'pypi' || inputs.MODE == 'extra') && inputs.PLATFORM == 'linux-x64' }}
         uses: actions/upload-artifact@v4
         with:
           name: rerun_notebook_wheel