diff --git a/action.yaml b/action.yaml
index d9470c9..d40e76b 100644
--- a/action.yaml
+++ b/action.yaml
@@ -372,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 }} == 'infer' ]]; then
+            if [[ $file != *"dags/"* ]]; then
               echo $file is not part of dags folder, triggering a image deploy
               DAGS_ONLY_DEPLOY=false
               break