diff --git a/mediapipe/framework/scheduler.cc b/mediapipe/framework/scheduler.cc
index ceadce78..23dc684c 100644
--- a/mediapipe/framework/scheduler.cc
+++ b/mediapipe/framework/scheduler.cc
@@ -270,13 +270,6 @@ absl::Status Scheduler::WaitForObservedOutput() {
   return observed ? absl::OkStatus() : absl::OutOfRangeError("Graph is done.");
 }
 
-// Idleness requires:
-// 1. either the graph has no source nodes or all source nodes are closed, and
-// 2. no packets are added to graph input streams.
-// For simplicity, we only fully support WaitUntilIdle() to be called on a graph
-// with no source nodes.
-// The application must ensure no other threads are adding packets to graph
-// input streams while a WaitUntilIdle() call is in progress.
 absl::Status Scheduler::WaitUntilIdle() {
   RET_CHECK_NE(state_, STATE_NOT_STARTED);
   ApplicationThreadAwait(std::bind(&Scheduler::IsIdle, this));
diff --git a/mediapipe/framework/scheduler.h b/mediapipe/framework/scheduler.h
index 8a6d079e..22d552c7 100644
--- a/mediapipe/framework/scheduler.h
+++ b/mediapipe/framework/scheduler.h
@@ -76,6 +76,16 @@ class Scheduler {
   // be scheduled and nothing is running in the worker threads.  This function
   // can be called only after Start().
   // Runs application thread tasks while waiting.
+  //
+  // Idleness requires:
+  // 1. either the graph has no source nodes or all source nodes are closed, and
+  // 2. no packets are added to graph input streams.
+  //
+  // For simplicity, we only fully support WaitUntilIdle() to be called on a
+  // graph with no source nodes.
+  //
+  // The application must ensure no other threads are adding packets to graph
+  // input streams while a WaitUntilIdle() call is in progress.
   absl::Status WaitUntilIdle() ABSL_LOCKS_EXCLUDED(state_mutex_);
 
   // Wait until any graph input stream has been unthrottled.