diff --git a/server/src/internalClusterTest/java/org/elasticsearch/rest/StreamingXContentResponseIT.java b/server/src/internalClusterTest/java/org/elasticsearch/rest/StreamingXContentResponseIT.java
index c76bfbfb4ea05..379370a8d31fb 100644
--- a/server/src/internalClusterTest/java/org/elasticsearch/rest/StreamingXContentResponseIT.java
+++ b/server/src/internalClusterTest/java/org/elasticsearch/rest/StreamingXContentResponseIT.java
@@ -171,11 +171,11 @@ private static void handleStreamingXContentRestRequest(
                 ThrottledIterator.run(
                     fragmentIterator,
                     (ref, fragment) -> randomFrom(EsExecutors.DIRECT_EXECUTOR_SERVICE, threadPool.generic()).execute(
-                        ActionRunnable.run(ActionListener.releaseAfter(refs.acquireListener(), ref), () -> {
+                        ActionRunnable.wrap(ActionListener.releaseAfter(refs.acquireListener(), ref), l -> {
                             Thread.yield();
                             streamingXContentResponse.writeFragment(
                                 p -> ChunkedToXContentHelper.chunk((b, xp) -> b.field(fragment.getKey(), fragment.getValue())),
-                                refs.acquire()
+                                () -> l.onResponse(null)
                             );
                         })
                     ),