diff --git a/action.yml b/action.yml
index 850af2c..15b169b 100644
--- a/action.yml
+++ b/action.yml
@@ -5,6 +5,9 @@ inputs:
   DOTNET_VERSION:
     description: '使用的 .NET 版本'
     default: '10.0'
+  DOTNET_ENVIRONMENT:
+    description: '使用的 .NET 環境名稱'
+    default: 'Production'
   PROJECT_NAME:
     description: '專案名稱(包含 Microsoft.EntityFrameworkCore.Design 的專案)'
     required: true
@@ -16,8 +19,6 @@ inputs:
     required: true
 runs:
   using: 'composite'
-  env:
-    ASPNETCORE_ENVIRONMENT: "Production"
   steps:
   - name: 取得專案
     uses: actions/checkout@v6
@@ -26,5 +27,8 @@ runs:
     with:
       dotnet-version: ${{ inputs.DOTNET_VERSION }}
   - name: 資料庫移轉
+    env:
+      ASPNETCORE_ENVIRONMENT: ${{ inputs.DOTNET_ENVIRONMENT }}
     run: dotnet ef database update --project ../${{ inputs.MIGRATION_PROJECT_NAME }} --context ${{ inputs.MIGRATION_CONTEXT }}
+    shell: bash
     working-directory: ${{ inputs.PROJECT_NAME }}
\ No newline at end of file