diff --git a/.github/workflows/publish-v2.yml b/.github/workflows/publish-v2.yml
index 17dc9d330..67ce0bdb2 100644
--- a/.github/workflows/publish-v2.yml
+++ b/.github/workflows/publish-v2.yml
@@ -1,9 +1,16 @@
 name: Publish Package v2
-on: [workflow_call]
-
+on:
+  workflow_call:
+    inputs:
+      environment:
+        type: string
+        required: true
+        description: 'The environment to publish to'
+        default: 'staging'
 jobs:
     publish:
         runs-on: ubuntu-latest
+        environment: ${{ inputs.environment }}
         steps:
             - uses: actions/checkout@v3