diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index 213988c..d744d4d 100644
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -11,6 +11,9 @@ jobs:
   publish:
     runs-on: ubuntu-latest
     if: github.ref_name == 'main' || startsWith(github.ref_name, 'release/')
+    permissions:
+      contents: read
+      packages: write
 
     steps:
       - uses: actions/checkout@v6.0.3
@@ -38,19 +41,6 @@ jobs:
         with:
           password: ${{ secrets.PYPI_API_TOKEN }}
 
-  publish-docker:
-    needs: publish
-    runs-on: ubuntu-latest
-    if: github.ref_name == 'main' || startsWith(github.ref_name, 'release/')
-    permissions:
-      contents: read
-      packages: write
-
-    steps:
-      - uses: actions/checkout@v6.0.3
-        with:
-          fetch-depth: 0
-
       - name: Log in to GitHub Container Registry
         uses: docker/login-action@v3
         with:
@@ -74,4 +64,6 @@ jobs:
           context: .
           push: true
           tags: ${{ steps.meta.outputs.tags }}
-          labels: ${{ steps.meta.outputs.labels }}
\ No newline at end of file
+          labels: ${{ steps.meta.outputs.labels }}
+          build-args: |
+            WHEEL=dist/*.whl
\ No newline at end of file