diff --git a/.github/workflows/deploy.yaml b/.github/workflows/deploy.yaml
index 1901105..6b679fd 100644
--- a/.github/workflows/deploy.yaml
+++ b/.github/workflows/deploy.yaml
@@ -17,7 +17,7 @@ on:
 
 jobs:
   deploy:
-    uses: mocbotau/infra-workflows/.github/workflows/generic-deploy.yaml@v0.3.0
+    uses: mocbotau/infra-workflows/.github/workflows/generic-deploy.yaml@v0.4.0
     with:
       event-name: ${{ github.event_name }}
       environment: ${{ github.event.inputs.env }}
diff --git a/.github/workflows/helmfile-diff.yaml b/.github/workflows/helmfile-diff.yaml
index c5f7a76..a504209 100644
--- a/.github/workflows/helmfile-diff.yaml
+++ b/.github/workflows/helmfile-diff.yaml
@@ -5,7 +5,9 @@ on:
 
 jobs:
   helmfile-diff:
-    uses: mocbotau/infra-workflows/.github/workflows/generic-helmfile-diff.yaml@v0.3.0
+    uses: mocbotau/infra-workflows/.github/workflows/generic-helmfile-diff.yaml@v0.4.0
+    with:
+      run-staging: true
     secrets: inherit
 
 permissions: