diff --git a/action.yml b/action.yml
index 9b01491..b99d8d0 100644
--- a/action.yml
+++ b/action.yml
@@ -38,7 +38,11 @@ runs:
       shell: bash
       run: |
         echo "::group::Running go mod vendor"
-        go mod vendor
+        if [[ -f go.work ]]; then
+          go work vendor
+        else
+          go mod vendor
+        fi
         echo "::endgroup::"
 
     - name: Create a deps tarball