From d2abf86648259b3e87f0513a6e2b19178be4c3e6 Mon Sep 17 00:00:00 2001
From: pghagare <pghagare@amazon.com>
Date: Mon, 8 Nov 2021 15:18:10 -0800
Subject: [PATCH] Update PubSubStress.java

---
 .../PubSubStress/src/main/java/pubsubstress/PubSubStress.java   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aws-iot-device-sdk-java-v2-main/samples/PubSubStress/src/main/java/pubsubstress/PubSubStress.java b/aws-iot-device-sdk-java-v2-main/samples/PubSubStress/src/main/java/pubsubstress/PubSubStress.java
index 9c918dc09..2b7c45e0e 100644
--- a/aws-iot-device-sdk-java-v2-main/samples/PubSubStress/src/main/java/pubsubstress/PubSubStress.java
+++ b/aws-iot-device-sdk-java-v2-main/samples/PubSubStress/src/main/java/pubsubstress/PubSubStress.java
@@ -384,7 +384,7 @@ public static void main(String[] args) {
                     }
 
                     for (CompletableFuture<Integer> publishFuture : publishFutures) {
-                        publishFuture.get();
+                        publishFuture.get(15, TimeUnit.SECONDS);
                     }
 
                     System.out.println("zzzzz");