diff --git a/tests/contrib/botocore/test.py b/tests/contrib/botocore/test.py
index bd3120e529b..af3fd38701d 100644
--- a/tests/contrib/botocore/test.py
+++ b/tests/contrib/botocore/test.py
@@ -1605,7 +1605,11 @@ def test_sns_send_message_batch_trace_injection_with_max_message_attributes(self
         spans = self.get_spans()
 
         # get SNS messages via SQS
-        response = sqs.receive_message(QueueUrl=queue["QueueUrl"], MessageAttributeNames=["_datadog"])
+        response = sqs.receive_message(
+            QueueUrl=queue["QueueUrl"],
+            MessageAttributeNames=["_datadog"],
+            WaitTimeSeconds=10,
+        )
 
         # clean up resources
         sqs.delete_queue(QueueUrl=sqs_url)
@@ -1626,6 +1630,7 @@ def test_sns_send_message_batch_trace_injection_with_max_message_attributes(self
         assert trace_json is None
 
         # receive message using SQS and ensure headers are present
+        assert response.get("Messages"), response
         assert len(response["Messages"]) == 1
         msg = response["Messages"][0]
         assert msg is not None