diff --git a/samples/js/photos/src/photo-set-iterator.js b/samples/js/photos/src/photo-set-iterator.js
index d4dc099736..09bcb7fb93 100644
--- a/samples/js/photos/src/photo-set-iterator.js
+++ b/samples/js/photos/src/photo-set-iterator.js
@@ -123,6 +123,17 @@ export class SinglePhotoSetIterator extends PhotoSetIterator {
     this._innerIter = startPhoto ? innerSet.iteratorAt(startPhoto) : innerSet.iterator();
     return false;
   }
+
+  async return(): Promise<AsyncIteratorResult<IterValue>> {
+    this._done = true;
+    if (this._innerIter) {
+      await this._innerIter.return();
+    }
+    if (this._outerIter) {
+      await this._outerIter.return();
+    }
+    return {done: true};
+  }
 }
 
 export class PhotoSetIntersectionIterator extends PhotoSetIterator {
@@ -182,6 +193,9 @@ export class EmptyIterator extends PhotoSetIterator {
   next(): Promise<AsyncIteratorResult<IterValue>> {
     return Promise.resolve({done: true});
   }
+  return(): Promise<AsyncIteratorResult<IterValue>> {
+    return Promise.resolve({done: true});
+  }
 }
 
 function isIterationOrder(r1: IterValue, r2: IterValue): boolean {