diff --git a/.github/workflows/deploy.yaml b/.github/workflows/deploy.yaml
index 5411853..4545cbc 100644
--- a/.github/workflows/deploy.yaml
+++ b/.github/workflows/deploy.yaml
@@ -7,7 +7,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: "prod"
diff --git a/.github/workflows/helmfile-diff.yaml b/.github/workflows/helmfile-diff.yaml
index cb551f9..065ea47 100644
--- a/.github/workflows/helmfile-diff.yaml
+++ b/.github/workflows/helmfile-diff.yaml
@@ -5,9 +5,7 @@ on:
 
 jobs:
   helmfile-diff:
-    uses: mocbotau/infra-workflows/.github/workflows/generic-helmfile-diff.yaml@v0.3.0
-    with:
-      environment: "prod"
+    uses: mocbotau/infra-workflows/.github/workflows/generic-helmfile-diff.yaml@v0.4.0
     secrets: inherit
 
 permissions: