diff --git a/.gitea/workflows/docker-publish.yml b/.gitea/workflows/docker-publish.yml
index ab419af..c339642 100644
--- a/.gitea/workflows/docker-publish.yml
+++ b/.gitea/workflows/docker-publish.yml
@@ -90,19 +90,15 @@ jobs:
         run: |
           COMMIT_SHA="${GITHUB_SHA:-$GITEA_SHA}"
           SHORT_SHA="$(echo "${COMMIT_SHA}" | cut -c1-12)"
-          
+
           DEPLOYMENT_NAME="dataprovider01-dataprovider-orchestration-platform-dagster-user-template-code-location"
-          
-          # Patch the deployment to update the image tag
+
+          # Keep both the code server image and DAGSTER_CURRENT_IMAGE in sync.
           kubectl patch deployment "${DEPLOYMENT_NAME}" \
             -n "${K8S_NAMESPACE}" \
-            --type='json' \
-            -p="[{'op': 'replace', 'path': '/spec/template/spec/containers/0/image', 'value':'${IMAGE_REPO}:${SHORT_SHA}'}]"
-          
-          # Trigger rollout restart to pick up new image
-          kubectl rollout restart deployment/"${DEPLOYMENT_NAME}" \
-            -n "${K8S_NAMESPACE}"
-          
+            --type='strategic' \
+            -p="{\"spec\":{\"template\":{\"spec\":{\"containers\":[{\"name\":\"dagster-user-deployments\",\"image\":\"${IMAGE_REPO}:${SHORT_SHA}\",\"env\":[{\"name\":\"DAGSTER_CURRENT_IMAGE\",\"value\":\"${IMAGE_REPO}:${SHORT_SHA}\"}]}]}}}}"
+
           # Wait for rollout to complete
           kubectl rollout status deployment/"${DEPLOYMENT_NAME}" \
             -n "${K8S_NAMESPACE}" \