address the comment

Signed-off-by: Ryan Leung <[email protected]>
tikv · Nov 15, 2023 · 52858a1 · 52858a1
1 parent 7c61679
commit 52858a1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/schedule/schedulers/scheduler_controller.go b/pkg/schedule/schedulers/scheduler_controller.go
@@ -72,8 +72,8 @@ func NewController(ctx context.Context, cluster sche.SchedulerCluster, storage e
 
 // Wait waits on all schedulers to exit.
 func (c *Controller) Wait() {
-	c.RLock()
-	defer c.RUnlock()
+	c.Lock()
+	defer c.Unlock()
 	c.wg.Wait()
 }