diff --git a/samples/automation_lambdas/EDR_API/health_check.py b/samples/automation_lambdas/EDR_API/health_check.py
index f96478d7b..c034e5577 100755
--- a/samples/automation_lambdas/EDR_API/health_check.py
+++ b/samples/automation_lambdas/EDR_API/health_check.py
@@ -1,5 +1,6 @@
 import json
 import boto3
+from botocore.exceptions import ClientError
 from falconpy import APIHarness
 
 # Function to grab secrets
diff --git a/samples/automation_lambdas/EDR_API/lambda_function.py b/samples/automation_lambdas/EDR_API/lambda_function.py
index 3c29e9ff8..20c48e0bd 100755
--- a/samples/automation_lambdas/EDR_API/lambda_function.py
+++ b/samples/automation_lambdas/EDR_API/lambda_function.py
@@ -1,5 +1,7 @@
 import json
 import boto3
+from botocore.exceptions import ClientError
+
 
 def get_secret(key_type):
     # Create a Secrets Manager client
diff --git a/samples/automation_lambdas/FalconBot-SlackEnd.py b/samples/automation_lambdas/FalconBot-SlackEnd.py
index 7cac0f8bb..cf728fc0a 100755
--- a/samples/automation_lambdas/FalconBot-SlackEnd.py
+++ b/samples/automation_lambdas/FalconBot-SlackEnd.py
@@ -3,6 +3,7 @@
 import hmac
 import hashlib
 import boto3
+from botocore.exceptions import ClientError
 from urllib.parse import parse_qs
 
 def get_secret():