diff --git a/dd-java-agent/agent-bootstrap/src/main/java/datadog/trace/bootstrap/instrumentation/buffer/InjectingPipeOutputStream.java b/dd-java-agent/agent-bootstrap/src/main/java/datadog/trace/bootstrap/instrumentation/buffer/InjectingPipeOutputStream.java
index 8fa6e115c4d..93bb8f956fe 100644
--- a/dd-java-agent/agent-bootstrap/src/main/java/datadog/trace/bootstrap/instrumentation/buffer/InjectingPipeOutputStream.java
+++ b/dd-java-agent/agent-bootstrap/src/main/java/datadog/trace/bootstrap/instrumentation/buffer/InjectingPipeOutputStream.java
@@ -2,6 +2,7 @@
 
 import java.io.IOException;
 import java.io.OutputStream;
+import javax.annotation.concurrent.NotThreadSafe;
 
 /**
  * An OutputStream containing a circular buffer with a lookbehind buffer of n bytes. The first time
@@ -9,6 +10,7 @@
  * thrown by the downstream, the buffer will be lost unless the error occurred when draining it. In
  * this case the draining will be resumed.
  */
+@NotThreadSafe
 public class InjectingPipeOutputStream extends OutputStream {
   private final byte[] lookbehind;
   private int pos;
@@ -168,6 +170,13 @@ private void drain() throws IOException {
     }
   }
 
+  public void commit() throws IOException {
+    if (filter || wasDraining) {
+      filter = false;
+      drain();
+    }
+  }
+
   @Override
   public void flush() throws IOException {
     downstream.flush();
@@ -176,9 +185,7 @@ public void flush() throws IOException {
   @Override
   public void close() throws IOException {
     try {
-      if (filter || wasDraining) {
-        drain();
-      }
+      commit();
     } finally {
       downstream.close();
     }
diff --git a/dd-java-agent/agent-bootstrap/src/main/java/datadog/trace/bootstrap/instrumentation/buffer/InjectingPipeWriter.java b/dd-java-agent/agent-bootstrap/src/main/java/datadog/trace/bootstrap/instrumentation/buffer/InjectingPipeWriter.java
index d7128e9b385..e435b23d132 100644
--- a/dd-java-agent/agent-bootstrap/src/main/java/datadog/trace/bootstrap/instrumentation/buffer/InjectingPipeWriter.java
+++ b/dd-java-agent/agent-bootstrap/src/main/java/datadog/trace/bootstrap/instrumentation/buffer/InjectingPipeWriter.java
@@ -2,6 +2,7 @@
 
 import java.io.IOException;
 import java.io.Writer;
+import javax.annotation.concurrent.NotThreadSafe;
 
 /**
  * A Writer containing a circular buffer with a lookbehind buffer of n bytes. The first time that
@@ -9,6 +10,7 @@
  * thrown by the downstream, the buffer will be lost unless the error occurred when draining it. In
  * this case the draining will be resumed.
  */
+@NotThreadSafe
 public class InjectingPipeWriter extends Writer {
   private final char[] lookbehind;
   private int pos;
@@ -169,6 +171,13 @@ private void drain() throws IOException {
     }
   }
 
+  public void commit() throws IOException {
+    if (filter || wasDraining) {
+      filter = false;
+      drain();
+    }
+  }
+
   @Override
   public void flush() throws IOException {
     downstream.flush();
@@ -177,9 +186,7 @@ public void flush() throws IOException {
   @Override
   public void close() throws IOException {
     try {
-      if (filter || wasDraining) {
-        drain();
-      }
+      commit();
     } finally {
       downstream.close();
     }
diff --git a/dd-java-agent/instrumentation/servlet/request-3/src/main/java/datadog/trace/instrumentation/servlet3/RumHttpServletResponseWrapper.java b/dd-java-agent/instrumentation/servlet/request-3/src/main/java/datadog/trace/instrumentation/servlet3/RumHttpServletResponseWrapper.java
index d7b173c6827..9ba84665901 100644
--- a/dd-java-agent/instrumentation/servlet/request-3/src/main/java/datadog/trace/instrumentation/servlet3/RumHttpServletResponseWrapper.java
+++ b/dd-java-agent/instrumentation/servlet/request-3/src/main/java/datadog/trace/instrumentation/servlet3/RumHttpServletResponseWrapper.java
@@ -2,18 +2,37 @@
 
 import datadog.trace.api.rum.RumInjector;
 import datadog.trace.bootstrap.instrumentation.buffer.InjectingPipeWriter;
+import datadog.trace.util.MethodHandles;
 import java.io.IOException;
 import java.io.PrintWriter;
+import java.lang.invoke.MethodHandle;
 import java.nio.charset.Charset;
 import javax.servlet.ServletOutputStream;
+import javax.servlet.ServletResponse;
 import javax.servlet.http.HttpServletResponse;
 import javax.servlet.http.HttpServletResponseWrapper;
 
 public class RumHttpServletResponseWrapper extends HttpServletResponseWrapper {
   private final RumInjector rumInjector;
-  private ServletOutputStream outputStream;
+  private WrappedServletOutputStream outputStream;
   private PrintWriter printWriter;
-  private boolean shouldInject = false;
+  private InjectingPipeWriter wrappedPipeWriter;
+  private boolean shouldInject = true;
+
+  private static final MethodHandle SET_CONTENT_LENGTH_LONG = getMh("setContentLengthLong");
+
+  private static MethodHandle getMh(final String name) {
+    try {
+      return new MethodHandles(ServletResponse.class.getClassLoader())
+          .method(ServletResponse.class, name);
+    } catch (Throwable ignored) {
+      return null;
+    }
+  }
+
+  private static <E extends Throwable> void sneakyThrow(Throwable e) throws E {
+    throw (E) e;
+  }
 
   public RumHttpServletResponseWrapper(HttpServletResponse response) {
     super(response);
@@ -22,50 +41,68 @@ public RumHttpServletResponseWrapper(HttpServletResponse response) {
 
   @Override
   public ServletOutputStream getOutputStream() throws IOException {
+    if (outputStream != null) {
+      return outputStream;
+    }
     if (!shouldInject) {
       return super.getOutputStream();
     }
-    if (outputStream == null) {
-      String encoding = getCharacterEncoding();
-      if (encoding == null) {
-        encoding = Charset.defaultCharset().name();
-      }
-      outputStream =
-          new WrappedServletOutputStream(
-              super.getOutputStream(),
-              rumInjector.getMarkerBytes(encoding),
-              rumInjector.getSnippetBytes(encoding),
-              this::onInjected);
+    String encoding = getCharacterEncoding();
+    if (encoding == null) {
+      encoding = Charset.defaultCharset().name();
     }
+    outputStream =
+        new WrappedServletOutputStream(
+            super.getOutputStream(),
+            rumInjector.getMarkerBytes(encoding),
+            rumInjector.getSnippetBytes(encoding),
+            this::onInjected);
+
     return outputStream;
   }
 
   @Override
   public PrintWriter getWriter() throws IOException {
-    final PrintWriter delegate = super.getWriter();
-    if (!shouldInject) {
-      return delegate;
+    if (printWriter != null) {
+      return printWriter;
     }
-    if (printWriter == null) {
-      printWriter =
-          new PrintWriter(
-              new InjectingPipeWriter(
-                  delegate,
-                  rumInjector.getMarkerChars(),
-                  rumInjector.getSnippetChars(),
-                  this::onInjected));
+    if (!shouldInject) {
+      return super.getWriter();
     }
+    wrappedPipeWriter =
+        new InjectingPipeWriter(
+            super.getWriter(),
+            rumInjector.getMarkerChars(),
+            rumInjector.getSnippetChars(),
+            this::onInjected);
+    printWriter = new PrintWriter(wrappedPipeWriter);
+
     return printWriter;
   }
 
   @Override
   public void setContentLength(int len) {
     // don't set it since we don't know if we will inject
+    if (!shouldInject) {
+      super.setContentLength(len);
+    }
+  }
+
+  @Override
+  public void setContentLengthLong(long len) {
+    if (!shouldInject && SET_CONTENT_LENGTH_LONG != null) {
+      try {
+        SET_CONTENT_LENGTH_LONG.invoke(getResponse(), len);
+      } catch (Throwable t) {
+        sneakyThrow(t);
+      }
+    }
   }
 
   @Override
   public void reset() {
     this.outputStream = null;
+    this.wrappedPipeWriter = null;
     this.printWriter = null;
     this.shouldInject = false;
     super.reset();
@@ -74,8 +111,8 @@ public void reset() {
   @Override
   public void resetBuffer() {
     this.outputStream = null;
+    this.wrappedPipeWriter = null;
     this.printWriter = null;
-    this.shouldInject = false;
     super.resetBuffer();
   }
 
@@ -89,7 +126,27 @@ public void onInjected() {
 
   @Override
   public void setContentType(String type) {
-    shouldInject = type != null && type.contains("text/html");
+    if (shouldInject) {
+      shouldInject = type != null && type.contains("text/html");
+    }
+    if (!shouldInject) {
+      commit();
+    }
     super.setContentType(type);
   }
+
+  public void commit() {
+    if (wrappedPipeWriter != null) {
+      try {
+        wrappedPipeWriter.commit();
+      } catch (Throwable ignored) {
+      }
+    }
+    if (outputStream != null) {
+      try {
+        outputStream.commit();
+      } catch (Throwable ignored) {
+      }
+    }
+  }
 }
diff --git a/dd-java-agent/instrumentation/servlet/request-3/src/main/java/datadog/trace/instrumentation/servlet3/Servlet3Advice.java b/dd-java-agent/instrumentation/servlet/request-3/src/main/java/datadog/trace/instrumentation/servlet3/Servlet3Advice.java
index 134d1f28229..3b1adba0938 100644
--- a/dd-java-agent/instrumentation/servlet/request-3/src/main/java/datadog/trace/instrumentation/servlet3/Servlet3Advice.java
+++ b/dd-java-agent/instrumentation/servlet/request-3/src/main/java/datadog/trace/instrumentation/servlet3/Servlet3Advice.java
@@ -35,7 +35,8 @@ public static boolean onEnter(
       @Advice.Argument(value = 1, readOnly = false) ServletResponse response,
       @Advice.Local("isDispatch") boolean isDispatch,
       @Advice.Local("finishSpan") boolean finishSpan,
-      @Advice.Local("contextScope") ContextScope scope) {
+      @Advice.Local("contextScope") ContextScope scope,
+      @Advice.Local("rumServletWrapper") RumHttpServletResponseWrapper rumServletWrapper) {
     final boolean invalidRequest =
         !(request instanceof HttpServletRequest) || !(response instanceof HttpServletResponse);
     if (invalidRequest) {
@@ -47,7 +48,8 @@ public static boolean onEnter(
 
     if (RumInjector.get().isEnabled() && httpServletRequest.getAttribute(DD_RUM_INJECTED) == null) {
       httpServletRequest.setAttribute(DD_RUM_INJECTED, Boolean.TRUE);
-      httpServletResponse = new RumHttpServletResponseWrapper(httpServletResponse);
+      rumServletWrapper = new RumHttpServletResponseWrapper(httpServletResponse);
+      httpServletResponse = rumServletWrapper;
       response = httpServletResponse;
     }
 
@@ -108,7 +110,11 @@ public static void stopSpan(
       @Advice.Local("contextScope") final ContextScope scope,
       @Advice.Local("isDispatch") boolean isDispatch,
       @Advice.Local("finishSpan") boolean finishSpan,
+      @Advice.Local("rumServletWrapper") RumHttpServletResponseWrapper rumServletWrapper,
       @Advice.Thrown final Throwable throwable) {
+    if (rumServletWrapper != null) {
+      rumServletWrapper.commit();
+    }
     // Set user.principal regardless of who created this span.
     final Object spanAttr = request.getAttribute(DD_SPAN_ATTRIBUTE);
     if (Config.get().isServletPrincipalEnabled()
diff --git a/dd-java-agent/instrumentation/servlet/request-3/src/main/java/datadog/trace/instrumentation/servlet3/WrappedServletOutputStream.java b/dd-java-agent/instrumentation/servlet/request-3/src/main/java/datadog/trace/instrumentation/servlet3/WrappedServletOutputStream.java
index d22d7899836..951eceb5db8 100644
--- a/dd-java-agent/instrumentation/servlet/request-3/src/main/java/datadog/trace/instrumentation/servlet3/WrappedServletOutputStream.java
+++ b/dd-java-agent/instrumentation/servlet/request-3/src/main/java/datadog/trace/instrumentation/servlet3/WrappedServletOutputStream.java
@@ -3,13 +3,12 @@
 import datadog.trace.bootstrap.instrumentation.buffer.InjectingPipeOutputStream;
 import datadog.trace.util.MethodHandles;
 import java.io.IOException;
-import java.io.OutputStream;
 import java.lang.invoke.MethodHandle;
 import javax.servlet.ServletOutputStream;
 import javax.servlet.WriteListener;
 
 public class WrappedServletOutputStream extends ServletOutputStream {
-  private final OutputStream filtered;
+  private final InjectingPipeOutputStream filtered;
   private final ServletOutputStream delegate;
 
   private static final MethodHandle IS_READY_MH = getMh("isReady");
@@ -83,4 +82,8 @@ public void setWriteListener(WriteListener writeListener) {
       sneakyThrow(e);
     }
   }
+
+  public void commit() throws IOException {
+    filtered.commit();
+  }
 }
diff --git a/dd-java-agent/instrumentation/servlet/request-3/src/testFixtures/groovy/datadog/trace/instrumentation/servlet3/RumServlet.groovy b/dd-java-agent/instrumentation/servlet/request-3/src/testFixtures/groovy/datadog/trace/instrumentation/servlet3/RumServlet.groovy
index 127950fc385..dd68e6223dc 100644
--- a/dd-java-agent/instrumentation/servlet/request-3/src/testFixtures/groovy/datadog/trace/instrumentation/servlet3/RumServlet.groovy
+++ b/dd-java-agent/instrumentation/servlet/request-3/src/testFixtures/groovy/datadog/trace/instrumentation/servlet3/RumServlet.groovy
@@ -14,8 +14,8 @@ class RumServlet extends HttpServlet {
 
   @Override
   protected void service(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
-    resp.setContentType(mimeType)
     try (def writer = resp.getWriter()) {
+      resp.setContentType(mimeType)
       writer.println("\n" +
       "<!doctype html>\n" +
       "<html>\n" +
diff --git a/dd-java-agent/instrumentation/servlet/request-5/src/main/java/datadog/trace/instrumentation/servlet5/JakartaServletInstrumentation.java b/dd-java-agent/instrumentation/servlet/request-5/src/main/java/datadog/trace/instrumentation/servlet5/JakartaServletInstrumentation.java
index 6ef4940a41e..91e3c989019 100644
--- a/dd-java-agent/instrumentation/servlet/request-5/src/main/java/datadog/trace/instrumentation/servlet5/JakartaServletInstrumentation.java
+++ b/dd-java-agent/instrumentation/servlet/request-5/src/main/java/datadog/trace/instrumentation/servlet5/JakartaServletInstrumentation.java
@@ -68,7 +68,8 @@ public static class JakartaServletAdvice {
     @Advice.OnMethodEnter(suppress = Throwable.class)
     public static AgentSpan before(
         @Advice.Argument(0) final ServletRequest request,
-        @Advice.Argument(value = 1, readOnly = false) ServletResponse response) {
+        @Advice.Argument(value = 1, readOnly = false) ServletResponse response,
+        @Advice.Local("rumServletWrapper") RumHttpServletResponseWrapper rumServletWrapper) {
       if (!(request instanceof HttpServletRequest)) {
         return null;
       }
@@ -79,7 +80,8 @@ public static AgentSpan before(
         if (RumInjector.get().isEnabled()
             && httpServletRequest.getAttribute(DD_RUM_INJECTED) == null) {
           httpServletRequest.setAttribute(DD_RUM_INJECTED, Boolean.TRUE);
-          response = new RumHttpServletResponseWrapper((HttpServletResponse) response);
+          rumServletWrapper = new RumHttpServletResponseWrapper((HttpServletResponse) response);
+          response = rumServletWrapper;
         }
       }
 
@@ -95,10 +97,15 @@ public static AgentSpan before(
 
     @Advice.OnMethodExit(onThrowable = Throwable.class, suppress = Throwable.class)
     public static void after(
-        @Advice.Enter final AgentSpan span, @Advice.Argument(0) final ServletRequest request) {
+        @Advice.Enter final AgentSpan span,
+        @Advice.Argument(0) final ServletRequest request,
+        @Advice.Local("rumServletWrapper") RumHttpServletResponseWrapper rumServletWrapper) {
       if (span == null) {
         return;
       }
+      if (rumServletWrapper != null) {
+        rumServletWrapper.commit();
+      }
 
       CallDepthThreadLocalMap.reset(HttpServletRequest.class);
       final HttpServletRequest httpServletRequest =
diff --git a/dd-java-agent/instrumentation/servlet/request-5/src/main/java/datadog/trace/instrumentation/servlet5/RumHttpServletResponseWrapper.java b/dd-java-agent/instrumentation/servlet/request-5/src/main/java/datadog/trace/instrumentation/servlet5/RumHttpServletResponseWrapper.java
index ce86c11e863..4b846b3c2db 100644
--- a/dd-java-agent/instrumentation/servlet/request-5/src/main/java/datadog/trace/instrumentation/servlet5/RumHttpServletResponseWrapper.java
+++ b/dd-java-agent/instrumentation/servlet/request-5/src/main/java/datadog/trace/instrumentation/servlet5/RumHttpServletResponseWrapper.java
@@ -11,9 +11,10 @@
 
 public class RumHttpServletResponseWrapper extends HttpServletResponseWrapper {
   private final RumInjector rumInjector;
-  private ServletOutputStream outputStream;
+  private WrappedServletOutputStream outputStream;
+  private InjectingPipeWriter wrappedPipeWriter;
   private PrintWriter printWriter;
-  private boolean shouldInject = false;
+  private boolean shouldInject = true;
 
   public RumHttpServletResponseWrapper(HttpServletResponse response) {
     super(response);
@@ -22,50 +23,63 @@ public RumHttpServletResponseWrapper(HttpServletResponse response) {
 
   @Override
   public ServletOutputStream getOutputStream() throws IOException {
+    if (outputStream != null) {
+      return outputStream;
+    }
     if (!shouldInject) {
       return super.getOutputStream();
     }
-    if (outputStream == null) {
-      String encoding = getCharacterEncoding();
-      if (encoding == null) {
-        encoding = Charset.defaultCharset().name();
-      }
-      outputStream =
-          new WrappedServletOutputStream(
-              super.getOutputStream(),
-              rumInjector.getMarkerBytes(encoding),
-              rumInjector.getSnippetBytes(encoding),
-              this::onInjected);
+    String encoding = getCharacterEncoding();
+    if (encoding == null) {
+      encoding = Charset.defaultCharset().name();
     }
+    outputStream =
+        new WrappedServletOutputStream(
+            super.getOutputStream(),
+            rumInjector.getMarkerBytes(encoding),
+            rumInjector.getSnippetBytes(encoding),
+            this::onInjected);
     return outputStream;
   }
 
   @Override
   public PrintWriter getWriter() throws IOException {
-    final PrintWriter delegate = super.getWriter();
-    if (!shouldInject) {
-      return delegate;
+    if (printWriter != null) {
+      return printWriter;
     }
-    if (printWriter == null) {
-      printWriter =
-          new PrintWriter(
-              new InjectingPipeWriter(
-                  delegate,
-                  rumInjector.getMarkerChars(),
-                  rumInjector.getSnippetChars(),
-                  this::onInjected));
+    if (!shouldInject) {
+      return super.getWriter();
     }
+    wrappedPipeWriter =
+        new InjectingPipeWriter(
+            super.getWriter(),
+            rumInjector.getMarkerChars(),
+            rumInjector.getSnippetChars(),
+            this::onInjected);
+    printWriter = new PrintWriter(wrappedPipeWriter);
+
     return printWriter;
   }
 
   @Override
   public void setContentLength(int len) {
     // don't set it since we don't know if we will inject
+    if (!shouldInject) {
+      super.setContentLength(len);
+    }
+  }
+
+  @Override
+  public void setContentLengthLong(long len) {
+    if (!shouldInject) {
+      super.setContentLengthLong(len);
+    }
   }
 
   @Override
   public void reset() {
     this.outputStream = null;
+    this.wrappedPipeWriter = null;
     this.printWriter = null;
     this.shouldInject = false;
     super.reset();
@@ -74,8 +88,8 @@ public void reset() {
   @Override
   public void resetBuffer() {
     this.outputStream = null;
+    this.wrappedPipeWriter = null;
     this.printWriter = null;
-    this.shouldInject = false;
     super.resetBuffer();
   }
 
@@ -89,7 +103,27 @@ public void onInjected() {
 
   @Override
   public void setContentType(String type) {
-    shouldInject = type != null && type.contains("text/html");
+    if (shouldInject) {
+      shouldInject = type != null && type.contains("text/html");
+    }
+    if (!shouldInject) {
+      commit();
+    }
     super.setContentType(type);
   }
+
+  public void commit() {
+    if (wrappedPipeWriter != null) {
+      try {
+        wrappedPipeWriter.commit();
+      } catch (Throwable ignored) {
+      }
+    }
+    if (outputStream != null) {
+      try {
+        outputStream.commit();
+      } catch (Throwable ignored) {
+      }
+    }
+  }
 }
diff --git a/dd-java-agent/instrumentation/servlet/request-5/src/main/java/datadog/trace/instrumentation/servlet5/WrappedServletOutputStream.java b/dd-java-agent/instrumentation/servlet/request-5/src/main/java/datadog/trace/instrumentation/servlet5/WrappedServletOutputStream.java
index 2c43af795f8..f2338ee1a71 100644
--- a/dd-java-agent/instrumentation/servlet/request-5/src/main/java/datadog/trace/instrumentation/servlet5/WrappedServletOutputStream.java
+++ b/dd-java-agent/instrumentation/servlet/request-5/src/main/java/datadog/trace/instrumentation/servlet5/WrappedServletOutputStream.java
@@ -4,10 +4,9 @@
 import jakarta.servlet.ServletOutputStream;
 import jakarta.servlet.WriteListener;
 import java.io.IOException;
-import java.io.OutputStream;
 
 public class WrappedServletOutputStream extends ServletOutputStream {
-  private final OutputStream filtered;
+  private final InjectingPipeOutputStream filtered;
   private final ServletOutputStream delegate;
 
   public WrappedServletOutputStream(
@@ -31,6 +30,10 @@ public void write(byte[] b, int off, int len) throws IOException {
     filtered.write(b, off, len);
   }
 
+  public void commit() throws IOException {
+    filtered.commit();
+  }
+
   @Override
   public void flush() throws IOException {
     filtered.flush();
diff --git a/dd-java-agent/instrumentation/servlet/request-5/src/testFixtures/groovy/datadog/trace/instrumentation/servlet5/RumServlet.groovy b/dd-java-agent/instrumentation/servlet/request-5/src/testFixtures/groovy/datadog/trace/instrumentation/servlet5/RumServlet.groovy
index 1fa57d02da8..af2851fda83 100644
--- a/dd-java-agent/instrumentation/servlet/request-5/src/testFixtures/groovy/datadog/trace/instrumentation/servlet5/RumServlet.groovy
+++ b/dd-java-agent/instrumentation/servlet/request-5/src/testFixtures/groovy/datadog/trace/instrumentation/servlet5/RumServlet.groovy
@@ -14,8 +14,8 @@ class RumServlet extends HttpServlet {
 
   @Override
   protected void service(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
-    resp.setContentType(mimeType)
     try (def writer = resp.getWriter()) {
+      resp.setContentType(mimeType)
       writer.println("\n" +
       "<!doctype html>\n" +
       "<html>\n" +
diff --git a/dd-smoke-tests/rum/src/main/groovy/datadog/smoketest/rum/AbstractRumServerSmokeTest.groovy b/dd-smoke-tests/rum/src/main/groovy/datadog/smoketest/rum/AbstractRumServerSmokeTest.groovy
index d9ece96fde9..a4b79227e36 100644
--- a/dd-smoke-tests/rum/src/main/groovy/datadog/smoketest/rum/AbstractRumServerSmokeTest.groovy
+++ b/dd-smoke-tests/rum/src/main/groovy/datadog/smoketest/rum/AbstractRumServerSmokeTest.groovy
@@ -50,11 +50,13 @@ class AbstractRumServerSmokeTest extends AbstractServerSmokeTest {
     assert response.header('x-datadog-rum-injected') == '1': 'RUM injected header missing'
     def content = response.body().string()
     assert content.contains('https://www.datadoghq-browser-agent.com'): 'RUM script not injected'
+    assert content.endsWith('</html>'): 'Response not fully flushed'
   }
 
   static void assertRumNotInjected(Response response) {
     assert response.header('x-datadog-rum-injected') == null: 'RUM header unexpectedly injected'
-    def content = response.body().toString()
+    def content = response.body().string()
     assert !content.contains('https://www.datadoghq-browser-agent.com'): 'RUM script unexpectedly injected'
+    assert content.endsWith('</response>'): 'Response not fully flushed'
   }
 }
diff --git a/dd-smoke-tests/rum/tomcat-10/src/main/java/com/example/HtmlServlet.java b/dd-smoke-tests/rum/tomcat-10/src/main/java/com/example/HtmlServlet.java
index 31a3c0825d9..0940634b377 100644
--- a/dd-smoke-tests/rum/tomcat-10/src/main/java/com/example/HtmlServlet.java
+++ b/dd-smoke-tests/rum/tomcat-10/src/main/java/com/example/HtmlServlet.java
@@ -9,21 +9,20 @@
 public class HtmlServlet extends HttpServlet {
   @Override
   protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
+    final PrintWriter writer = resp.getWriter();
     resp.setContentType("text/html;charset=UTF-8");
-    try (final PrintWriter writer = resp.getWriter()) {
-      writer.write(
-          "<!DOCTYPE html>"
-              + "<html lang=\"en\">"
-              + "<head>"
-              + "    <meta charset=\"UTF-8\">"
-              + "    <meta name=\"viewport\" content=\"width=device-width, initial-scale=1.0\">"
-              + "    <title>Hello Servlet</title>"
-              + "</head>"
-              + "<body>"
-              + "    <h1>Hello from Tomcat 9 Servlet!</h1>"
-              + "    <p>This is a demo HTML page served by Java servlet.</p>"
-              + "</body>"
-              + "</html>");
-    }
+    writer.write(
+        "<!DOCTYPE html>"
+            + "<html lang=\"en\">"
+            + "<head>"
+            + "    <meta charset=\"UTF-8\">"
+            + "    <meta name=\"viewport\" content=\"width=device-width, initial-scale=1.0\">"
+            + "    <title>Hello Servlet</title>"
+            + "</head>"
+            + "<body>"
+            + "    <h1>Hello from Tomcat 9 Servlet!</h1>"
+            + "    <p>This is a demo HTML page served by Java servlet.</p>"
+            + "</body>"
+            + "</html>");
   }
 }
diff --git a/dd-smoke-tests/rum/tomcat-10/src/main/java/com/example/XmlServlet.java b/dd-smoke-tests/rum/tomcat-10/src/main/java/com/example/XmlServlet.java
index 8f8399bb408..43cf679cc09 100644
--- a/dd-smoke-tests/rum/tomcat-10/src/main/java/com/example/XmlServlet.java
+++ b/dd-smoke-tests/rum/tomcat-10/src/main/java/com/example/XmlServlet.java
@@ -9,17 +9,16 @@
 public class XmlServlet extends HttpServlet {
   @Override
   protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
+    final PrintWriter out = resp.getWriter();
     resp.setContentType("application/xml;charset=UTF-8");
-    try (PrintWriter out = resp.getWriter()) {
-      out.println("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
-      out.println("<response>");
-      out.println("    <status>success</status>");
-      out.println("    <message>RUM injection test</message>");
-      out.println("    <data>");
-      out.println("        <item id=\"1\">Test Item 1</item>");
-      out.println("        <item id=\"2\">Test Item 2</item>");
-      out.println("    </data>");
-      out.println("</response>");
-    }
+    out.println("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
+    out.println("<response>");
+    out.println("    <status>success</status>");
+    out.println("    <message>RUM injection test</message>");
+    out.println("    <data>");
+    out.println("        <item id=\"1\">Test Item 1</item>");
+    out.println("        <item id=\"2\">Test Item 2</item>");
+    out.println("    </data>");
+    out.print("</response>");
   }
 }
diff --git a/dd-smoke-tests/rum/tomcat-11/src/main/java/com/example/HtmlServlet.java b/dd-smoke-tests/rum/tomcat-11/src/main/java/com/example/HtmlServlet.java
index 31a3c0825d9..0940634b377 100644
--- a/dd-smoke-tests/rum/tomcat-11/src/main/java/com/example/HtmlServlet.java
+++ b/dd-smoke-tests/rum/tomcat-11/src/main/java/com/example/HtmlServlet.java
@@ -9,21 +9,20 @@
 public class HtmlServlet extends HttpServlet {
   @Override
   protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
+    final PrintWriter writer = resp.getWriter();
     resp.setContentType("text/html;charset=UTF-8");
-    try (final PrintWriter writer = resp.getWriter()) {
-      writer.write(
-          "<!DOCTYPE html>"
-              + "<html lang=\"en\">"
-              + "<head>"
-              + "    <meta charset=\"UTF-8\">"
-              + "    <meta name=\"viewport\" content=\"width=device-width, initial-scale=1.0\">"
-              + "    <title>Hello Servlet</title>"
-              + "</head>"
-              + "<body>"
-              + "    <h1>Hello from Tomcat 9 Servlet!</h1>"
-              + "    <p>This is a demo HTML page served by Java servlet.</p>"
-              + "</body>"
-              + "</html>");
-    }
+    writer.write(
+        "<!DOCTYPE html>"
+            + "<html lang=\"en\">"
+            + "<head>"
+            + "    <meta charset=\"UTF-8\">"
+            + "    <meta name=\"viewport\" content=\"width=device-width, initial-scale=1.0\">"
+            + "    <title>Hello Servlet</title>"
+            + "</head>"
+            + "<body>"
+            + "    <h1>Hello from Tomcat 9 Servlet!</h1>"
+            + "    <p>This is a demo HTML page served by Java servlet.</p>"
+            + "</body>"
+            + "</html>");
   }
 }
diff --git a/dd-smoke-tests/rum/tomcat-11/src/main/java/com/example/XmlServlet.java b/dd-smoke-tests/rum/tomcat-11/src/main/java/com/example/XmlServlet.java
index 8f8399bb408..43cf679cc09 100644
--- a/dd-smoke-tests/rum/tomcat-11/src/main/java/com/example/XmlServlet.java
+++ b/dd-smoke-tests/rum/tomcat-11/src/main/java/com/example/XmlServlet.java
@@ -9,17 +9,16 @@
 public class XmlServlet extends HttpServlet {
   @Override
   protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
+    final PrintWriter out = resp.getWriter();
     resp.setContentType("application/xml;charset=UTF-8");
-    try (PrintWriter out = resp.getWriter()) {
-      out.println("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
-      out.println("<response>");
-      out.println("    <status>success</status>");
-      out.println("    <message>RUM injection test</message>");
-      out.println("    <data>");
-      out.println("        <item id=\"1\">Test Item 1</item>");
-      out.println("        <item id=\"2\">Test Item 2</item>");
-      out.println("    </data>");
-      out.println("</response>");
-    }
+    out.println("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
+    out.println("<response>");
+    out.println("    <status>success</status>");
+    out.println("    <message>RUM injection test</message>");
+    out.println("    <data>");
+    out.println("        <item id=\"1\">Test Item 1</item>");
+    out.println("        <item id=\"2\">Test Item 2</item>");
+    out.println("    </data>");
+    out.print("</response>");
   }
 }
diff --git a/dd-smoke-tests/rum/tomcat-9/src/main/java/com/example/HtmlServlet.java b/dd-smoke-tests/rum/tomcat-9/src/main/java/com/example/HtmlServlet.java
index ed7283ec81e..90f13e7317f 100644
--- a/dd-smoke-tests/rum/tomcat-9/src/main/java/com/example/HtmlServlet.java
+++ b/dd-smoke-tests/rum/tomcat-9/src/main/java/com/example/HtmlServlet.java
@@ -1,7 +1,6 @@
 package com.example;
 
 import java.io.IOException;
-import java.io.PrintWriter;
 import javax.servlet.http.HttpServlet;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
@@ -9,21 +8,19 @@
 public class HtmlServlet extends HttpServlet {
   @Override
   protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
-    resp.setContentType("text/html;charset=UTF-8");
-    try (final PrintWriter writer = resp.getWriter()) {
-      writer.write(
-          "<!DOCTYPE html>"
-              + "<html lang=\"en\">"
-              + "<head>"
-              + "    <meta charset=\"UTF-8\">"
-              + "    <meta name=\"viewport\" content=\"width=device-width, initial-scale=1.0\">"
-              + "    <title>Hello Servlet</title>"
-              + "</head>"
-              + "<body>"
-              + "    <h1>Hello from Tomcat 9 Servlet!</h1>"
-              + "    <p>This is a demo HTML page served by Java servlet.</p>"
-              + "</body>"
-              + "</html>");
-    }
+    resp.getWriter()
+        .write(
+            "<!DOCTYPE html>"
+                + "<html lang=\"en\">"
+                + "<head>"
+                + "    <meta charset=\"UTF-8\">"
+                + "    <meta name=\"viewport\" content=\"width=device-width, initial-scale=1.0\">"
+                + "    <title>Hello Servlet</title>"
+                + "</head>"
+                + "<body>"
+                + "    <h1>Hello from Tomcat 9 Servlet!</h1>"
+                + "    <p>This is a demo HTML page served by Java servlet.</p>"
+                + "</body>"
+                + "</html>");
   }
 }
diff --git a/dd-smoke-tests/rum/tomcat-9/src/main/java/com/example/XmlServlet.java b/dd-smoke-tests/rum/tomcat-9/src/main/java/com/example/XmlServlet.java
index 15bb275acac..8f7a02b7a6f 100644
--- a/dd-smoke-tests/rum/tomcat-9/src/main/java/com/example/XmlServlet.java
+++ b/dd-smoke-tests/rum/tomcat-9/src/main/java/com/example/XmlServlet.java
@@ -10,16 +10,15 @@ public class XmlServlet extends HttpServlet {
   @Override
   protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
     resp.setContentType("application/xml;charset=UTF-8");
-    try (PrintWriter out = resp.getWriter()) {
-      out.println("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
-      out.println("<response>");
-      out.println("    <status>success</status>");
-      out.println("    <message>RUM injection test</message>");
-      out.println("    <data>");
-      out.println("        <item id=\"1\">Test Item 1</item>");
-      out.println("        <item id=\"2\">Test Item 2</item>");
-      out.println("    </data>");
-      out.println("</response>");
-    }
+    final PrintWriter out = resp.getWriter();
+    out.println("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
+    out.println("<response>");
+    out.println("    <status>success</status>");
+    out.println("    <message>RUM injection test</message>");
+    out.println("    <data>");
+    out.println("        <item id=\"1\">Test Item 1</item>");
+    out.println("        <item id=\"2\">Test Item 2</item>");
+    out.println("    </data>");
+    out.print("</response>");
   }
 }