diff --git a/azure-pipelines-staging.yml b/azure-pipelines-staging.yml
index b8e341e..faef499 100644
--- a/azure-pipelines-staging.yml
+++ b/azure-pipelines-staging.yml
@@ -122,7 +122,7 @@ jobs:
     displayName: Bash Script
     inputs:
       targetType: inline
-      script: cd dictation_client && cp -f .env.$(environment) .env && npm run build
+      script: cd dictation_client && npm run build:stg
   - task: ArchiveFiles@2
     inputs:
       rootFolderOrFile: dictation_client/build
@@ -167,7 +167,7 @@ jobs:
     displayName: Bash Script
     inputs:
       targetType: inline
-      script: cd dictation_client && cp -f .env.$(environment) .env && npm run build
+      script: cd dictation_client && npm run build:prod
   - task: ArchiveFiles@2
     inputs:
       rootFolderOrFile: dictation_client/build
diff --git a/dictation_client/package.json b/dictation_client/package.json
index b2283f6..8da7f1a 100644
--- a/dictation_client/package.json
+++ b/dictation_client/package.json
@@ -7,9 +7,10 @@
   },
   "scripts": {
     "start": "vite",
-    "build": "tsc && vite build",
-    "build:prod": "tsc && vite build",
-    "build:local": "tsc && vite build && sh localdeploy.sh",
+    "build": "tsc && vite build -- --mode development",
+    "build:stg": "tsc && vite build -- --mode staging",
+    "build:prod": "tsc && vite build -- --mode production",
+    "build:local": "tsc && vite build -- --mode development && sh localdeploy.sh",
     "preview": "vite preview",
     "typecheck": "tsc --noEmit",
     "codegen": "sh codegen.sh",