diff --git a/.github/workflows/build_jlext.yml b/.github/workflows/build_jlext.yml
index dd7d625..9f0e720 100644
--- a/.github/workflows/build_jlext.yml
+++ b/.github/workflows/build_jlext.yml
@@ -4,14 +4,17 @@ on:
       extension-name:
         required: true
         type: string
-
+      dist-dir:
+          required: false
+          type: string
+          default: 'dist'
       artifact-name:
         required: false
         type: string
         default: 'artifact'
 
 jobs:
-  build_java_project:
+  build_jlext_project:
     runs-on: ubuntu_act_on_20
     # container:
     #   image: catthehacker/ubuntu:act-latest
@@ -64,11 +67,13 @@ jobs:
           pip install build
           python -m build
           pip uninstall -y "${{ inputs.extension-name }}" jupyterlab
+
+          mv ${{ inputs.dist-dir }} pub
   
       - uses: actions/upload-artifact@v3
         with:
           name: ${{ inputs.artifact-name }}
           path: |
-            dist
+            pub
             docker
           if-no-files-found: error