diff --git a/tfjs-data/src/readers.ts b/tfjs-data/src/readers.ts
index 172b524795c..7db1f05e1ec 100644
--- a/tfjs-data/src/readers.ts
+++ b/tfjs-data/src/readers.ts
@@ -170,7 +170,7 @@ export function func<T extends TensorContainer>(
  *   return iterator;
  * }
  * const ds = tf.data.generator(makeIterator);
- * ds.forEachAsync(e => console.log(e));
+ * await ds.forEachAsync(e => console.log(e));
  * ```
  *
  * Example of creating a dataset from a generator:
@@ -186,7 +186,7 @@ export function func<T extends TensorContainer>(
  * }
  *
  * const ds = tf.data.generator(dataGenerator);
- * ds.forEachAsync(e => console.log(e));
+ * await ds.forEachAsync(e => console.log(e));
  * ```
  *
  * @param generator A Javascript generator function that returns a JavaScript