diff --git a/file-server/Dockerfile b/file-server/Dockerfile
index c098eae5..77059741 100644
--- a/file-server/Dockerfile
+++ b/file-server/Dockerfile
@@ -26,4 +26,4 @@ LABEL org.label-schema.name="FlowFuse File Storage" \
 
 EXPOSE 3001
 
-CMD ["./node_modules/.bin/ff-file-storage"]
\ No newline at end of file
+CMD ["./node_modules/.bin/ff-file-storage"]
diff --git a/helm/flowforge/Chart.lock b/helm/flowforge/Chart.lock
index c10b7660..4a972786 100644
--- a/helm/flowforge/Chart.lock
+++ b/helm/flowforge/Chart.lock
@@ -2,5 +2,5 @@ dependencies:
 - name: postgresql
   repository: https://charts.bitnami.com/bitnami
   version: 11.9.13
-digest: sha256:ea11617d363e174146b33ceafb5e2072d36731c2446178a91e5e6f705d0db067
-generated: "2023-11-15T14:44:27.496269+01:00"
+digest: sha256:1423e37aa789fccab1a4b5db593ccda3887032de2a4814b928df938f6db1de3a
+generated: "2024-02-13T12:15:48.820039+01:00"
diff --git a/helm/flowforge/Chart.yaml b/helm/flowforge/Chart.yaml
index a2d49fd5..15dbc1af 100644
--- a/helm/flowforge/Chart.yaml
+++ b/helm/flowforge/Chart.yaml
@@ -13,6 +13,9 @@ dependencies:
     version: 11.9.13
     repository: https://charts.bitnami.com/bitnami
     condition: forge.localPostgresql
+    import-values:
+      - child: image
+        parent: postgresql.image
 maintainers:
   - name: "FlowForge Inc"
     url: "https://flowfuse.com"
diff --git a/helm/flowforge/templates/job-upgrade-db.yaml b/helm/flowforge/templates/job-upgrade-db.yaml
index ed084171..c5555b14 100644
--- a/helm/flowforge/templates/job-upgrade-db.yaml
+++ b/helm/flowforge/templates/job-upgrade-db.yaml
@@ -8,8 +8,8 @@ metadata:
 data:
   upgrade.sh: |
     #!/bin/sh
-    apk add --no-cache postgresql14-client
-    psql -v ON_ERROR_STOP=1 -h {{ include "forge.databaseHost" . }} -U postgres postgres <<-ESQL
+    # apk add --no-cache postgresql14-client
+    psql -v ON_ERROR_STOP=1 -h {{ .Release.Name }}-postgresql -U postgres postgres <<-ESQL
     SELECT datname FROM pg_database WHERE datistemplate = false;
     SELECT 'CREATE DATABASE "ff-context"' WHERE NOT EXISTS (SELECT FROM pg_database WHERE datname = 'ff-context')\gexec
     GRANT ALL PRIVILEGES ON DATABASE "ff-context" TO "forge";
@@ -35,7 +35,7 @@ spec:
       restartPolicy: Never
       containers:
       - name: add-context-db
-        image: "alpine:3.17.0"
+        image: {{ .Values.postgresql.image.registry }}/{{ .Values.postgresql.image.repository }}/{{ .Values.postgresql.image.tag }}
         env:
         - name: PGPASSWORD
           valueFrom: