diff --git a/.github/workflows/fileserver-container.yml b/.github/workflows/fileserver-container.yml
index c6e052f8..258f3e63 100644
--- a/.github/workflows/fileserver-container.yml
+++ b/.github/workflows/fileserver-container.yml
@@ -130,7 +130,7 @@ jobs:
   update-production-image:
     if: github.ref_name == 'main'
     name: Update production image
-    needs: upload-production-image
+    needs: [upload-production-image, update-stage-image]
     runs-on: ubuntu-latest
     environment: production
     steps:
diff --git a/.github/workflows/flowforge-container.yml b/.github/workflows/flowforge-container.yml
index 50534535..c83faf9b 100644
--- a/.github/workflows/flowforge-container.yml
+++ b/.github/workflows/flowforge-container.yml
@@ -132,7 +132,7 @@ jobs:
   update-production-image:
     if: github.ref_name == 'main'
     name: Update production image
-    needs: upload-production-image
+    needs: [upload-production-image, update-stage-image]
     runs-on: ubuntu-latest
     environment: production
     steps: