diff --git a/action.yaml b/action.yaml
index 598e8c7..d9470c9 100644
--- a/action.yaml
+++ b/action.yaml
@@ -351,7 +351,11 @@ runs:
         git fetch origin $branch
         files=$(git diff --name-only ${{ github.event.before }} ${{ github.event.after }})
         echo "files changed: $files"
+
         DAGS_ONLY_DEPLOY=true
+        if [[ ${{ inputs.deploy-type }} == 'image-and-dags' ]]; then
+          DAGS_ONLY_DEPLOY=false
+        fi
 
         SKIP_IMAGE_OR_DAGS_DEPLOY=true
 
@@ -368,7 +372,7 @@ runs:
           if [[ $file =~ ^"$root_folder".* ]]; then
             echo $file is part of the input root folder
             SKIP_IMAGE_OR_DAGS_DEPLOY=false
-            if [[ $file != *"dags/"* && ${{ inputs.deploy-type }} != 'dags-only' ]]; then
+            if [[ $file != *"dags/"* && ${{ inputs.deploy-type }} == 'infer' ]]; then
               echo $file is not part of dags folder, triggering a image deploy
               DAGS_ONLY_DEPLOY=false
               break