diff --git a/sdks/java/io/amazon-web-services2/src/test/java/org/apache/beam/sdk/io/aws2/common/AsyncBatchWriteHandlerTest.java b/sdks/java/io/amazon-web-services2/src/test/java/org/apache/beam/sdk/io/aws2/common/AsyncBatchWriteHandlerTest.java
index cd7aca3c9f9b..056d856e442c 100644
--- a/sdks/java/io/amazon-web-services2/src/test/java/org/apache/beam/sdk/io/aws2/common/AsyncBatchWriteHandlerTest.java
+++ b/sdks/java/io/amazon-web-services2/src/test/java/org/apache/beam/sdk/io/aws2/common/AsyncBatchWriteHandlerTest.java
@@ -230,7 +230,7 @@ public void correctlyLimitConcurrency() throws Throwable {
     nextResults.complete(emptyList());
 
     handler.waitForCompletion();
-    assertThat(future).isDone();
+    eventually(5, () -> assertThat(future).isDone());
   }
 
   static class SubmitFn<T, V> implements BiFunction<String, List<T>, CompletableFuture<List<V>>> {