diff --git a/.gitea/workflows/update.yml b/.gitea/workflows/update.yml
index db0ad07..7ac8212 100644
--- a/.gitea/workflows/update.yml
+++ b/.gitea/workflows/update.yml
@@ -92,6 +92,8 @@ jobs:
         tags: |
           type=schedule,pattern={{date 'YYYYMMDD'}}
           type=raw,value=latest
+        labels: |
+          base-image.digest=${{ needs.check-images.outputs.base_image_digest }}
     - name: Build and push Docker image
       id: build-publish
       uses: docker/build-push-action@v5
@@ -105,7 +107,6 @@ jobs:
         build-args: |
           BASE_IMAGE_DIGEST=${{ needs.check-images.outputs.base_image_digest }}
     - name: Update deployment info
-      if: steps.build-publish.outputs.success == 'true'
       run: |
         echo "Docker image built and pushed successfully at $(date)"
         echo "Image tags:"
diff --git a/Dockerfile b/Dockerfile
index c93a72e..36ce079 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,9 +1,7 @@
 FROM ghcr.io/mlflow/mlflow:latest
-ARG BASE_IMAGE_DIGEST=unknown
 
-# Set a label with the base image digest for traceability
-LABEL base-image.digest="${BASE_IMAGE_DIGEST}" \
-    build-date="$(date -u +'%Y-%m-%dT%H:%M:%SZ')" \
+# Set a labels for traceability
+LABEL build-date="$(date -u +'%Y-%m-%dT%H:%M:%SZ')" \
     maintainer="Brian Jensen"
 
 # Use --no-cache-dir to avoid storing pip cache in the image