diff --git a/code/lib/instrumenter/src/instrumenter.ts b/code/lib/instrumenter/src/instrumenter.ts
index 9d22175073a3..b8c9cce1c47a 100644
--- a/code/lib/instrumenter/src/instrumenter.ts
+++ b/code/lib/instrumenter/src/instrumenter.ts
@@ -503,7 +503,7 @@ export class Instrumenter {
           try {
             const res = arg(...args);
             if (res instanceof Promise) {
-              willRestore = true;
+              willRestore = true; // We need to wait for the promise to finish before restoring
               return res.finally(restore);
             }
             return res;