diff --git a/.github/workflows/copy-database.yml b/.github/workflows/copy-database.yml
index 53c5666..ec6a9f2 100644
--- a/.github/workflows/copy-database.yml
+++ b/.github/workflows/copy-database.yml
@@ -6,6 +6,10 @@ name: Copy Database
 
 on:
   workflow_call:
+    inputs:
+      environment:
+        required: true
+        type: string
     secrets:
       aws-oidc-role-arn:
         required: true
@@ -15,6 +19,7 @@ on:
 jobs:
   Export:
     runs-on: ubuntu-latest
+    environment: ${{ inputs.environment }}
     container:
       image: ghcr.io/toumoro/tm-ansible:latest
     permissions:
diff --git a/templates/database-export.yml b/templates/database-export.yml
index 4ba37ec..b28c9ac 100644
--- a/templates/database-export.yml
+++ b/templates/database-export.yml
@@ -12,6 +12,8 @@ permissions:
 jobs:
   DatabaseExport:
     uses: toumoro/tm-workflows/.github/workflows/copy-database.yml@v1
+    with:
+      environment: 'production'
     secrets: 
       aws-oidc-role-arn: ${{ secrets.TM_CLIENT_AWS_OIDC_ROLE_ARN }}
       tm-tmdt-aws-oidc-role-arn: ${{ secrets.TM_TMDT_AWS_OIDC_ROLE_ARN }}