diff --git a/.gitea/workflows/update.yml b/.gitea/workflows/update.yml
index c522535..6d0f4d7 100644
--- a/.gitea/workflows/update.yml
+++ b/.gitea/workflows/update.yml
@@ -88,17 +88,17 @@ jobs:
         if [ "${{ github.event.inputs.skip_image_check }}" == "true" ]; then
           echo "Image check skipped, forcing build"
           echo "should_build=true" >> $GITHUB_OUTPUT
-        elif [ "${{ steps.mlflow-base.outputs.success }}" == "true" ] && \
-           [ "${{ steps.mlflow-full.outputs.success }}" == "true" ]; then
+        elif [ "${{ needs.get-mlflow-digest.outputs.success }}" == "true" ] && \
+           [ "${{ needs.get-mlflow-full-base-image-digest.outputs.success }}" == "true" ]; then
           echo "✅ Both MLflow base and MLflow-full images found"
-          if [ "${{ steps.mlflow-base.outputs.digest }}" != "${{ steps.mlflow-full.outputs.base_image_digest }}" ]; then
+          if [ "${{ needs.get-mlflow-digest.outputs.digest }}" != "${{ needs.get-mlflow-full-base-image-digest.outputs.base_image_digest }}" ]; then
             echo "Base image digest has changed"
             echo "should_build=true" >> $GITHUB_OUTPUT
           else
             echo "Base image digest is unchanged"
             echo "should_build=false" >> $GITHUB_OUTPUT
           fi
-        elif [ "${{ steps.mlflow-base.outputs.success }}" != "true" ]; then
+        elif [ "${{ needs.get-mlflow-base.outputs.success }}" != "true" ]; then
           echo "❌ MLflow base image not found"
           echo "should_build=false" >> $GITHUB_OUTPUT
         else