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");