diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index a8e9a06..cd02028 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -110,6 +110,18 @@ jobs:
           debug: true
           publish: false
 
+      - name: check max attempts flag
+        uses: ./
+        with:
+          aws_access_key_id: ${{ secrets.AWS_ACCESS_KEY_ID }}
+          aws_secret_access_key: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
+          aws_region: ${{ secrets.AWS_REGION }}
+          function_name: gorush
+          zip_file: example/deployment.zip
+          debug: true
+          publish: false
+          max_attempts: 200
+
   # deploy_source:
   #   name: deploy lambda from source
   #   runs-on: ubuntu-latest
diff --git a/Dockerfile b/Dockerfile
index a5de98c..9274eb6 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM ghcr.io/appleboy/drone-lambda:1.3.4
+FROM ghcr.io/appleboy/drone-lambda:1.3.5
 
 COPY entrypoint.sh /entrypoint.sh
 RUN chmod +x /entrypoint.sh
diff --git a/action.yml b/action.yml
index 4ebd12a..05f197a 100644
--- a/action.yml
+++ b/action.yml
@@ -60,6 +60,9 @@ inputs:
     description: 'A description of the function.'
   tracing_mode:
     description: 'Set Mode to Active to sample and trace a subset of incoming requests with AWS X-Ray.'
+  max_attempts:
+    description: 'the maximum number of times the waiter should attempt to check the resource for the target state'
+    default: 300
 
 runs:
   using: 'docker'