diff --git a/doc/api/stream.md b/doc/api/stream.md
index cf03163218bfbc..82cf99b4332000 100644
--- a/doc/api/stream.md
+++ b/doc/api/stream.md
@@ -2682,12 +2682,22 @@ const finished = util.promisify(stream.finished);
 
 const writable = fs.createWriteStream('./file');
 
+function drain(writable) {
+  if (writable.destroyed) {
+    return Promise.reject(new Error('premature close'));
+  }
+  return Promise.race([
+    once(writable, 'drain'),
+    once(writable, 'close')
+      .then(() => Promise.reject(new Error('premature close')))
+  ]);
+}
+
 async function pump(iterable, writable) {
   for await (const chunk of iterable) {
     // Handle backpressure on write().
     if (!writable.write(chunk)) {
-      if (writable.destroyed) return;
-      await once(writable, 'drain');
+      await drain(writable);
     }
   }
   writable.end();