diff --git a/aws-iot-device-sdk-java-v2-main/samples/Identity/src/main/java/identity/FleetProvisioningSample.java b/aws-iot-device-sdk-java-v2-main/samples/Identity/src/main/java/identity/FleetProvisioningSample.java
index cc21ca6da..8c465c154 100644
--- a/aws-iot-device-sdk-java-v2-main/samples/Identity/src/main/java/identity/FleetProvisioningSample.java
+++ b/aws-iot-device-sdk-java-v2-main/samples/Identity/src/main/java/identity/FleetProvisioningSample.java
@@ -301,7 +301,7 @@ private static void createKeysAndCertificateWorkflow() throws Exception {
                 FleetProvisioningSample::onRejectedRegister,
                 FleetProvisioningSample::onException);
 
-        subscribedRegisterRejected.get();
+        subscribedRegisterRejected.get(15, TimeUnit.SECONDS);
         System.out.println("Subscribed to SubscribeToRegisterThingRejected");
 
         CompletableFuture<Integer> publishKeys = iotIdentityClient.PublishCreateKeysAndCertificate(