1 files changed, 3 insertions, 1 deletions
diff --git a/src/mongo/db/s/balancer/cluster_chunks_resize_policy_impl.h b/src/mongo/db/s/balancer/cluster_chunks_resize_policy_impl.h
index 7bdcdd79596..b53b7afbeda 100644
--- a/src/mongo/db/s/balancer/cluster_chunks_resize_policy_impl.h
+++ b/src/mongo/db/s/balancer/cluster_chunks_resize_policy_impl.h
@@ -57,7 +57,7 @@ public:
 
     void actionCompleted(std::vector<ActionRequestInfo>&& followUpRequests);
 
-    void errorDetectedOnActionCompleted();
+    void errorDetectedOnActionCompleted(OperationContext* opCtx);
 
     bool restartNeeded() const;
 
@@ -86,6 +86,8 @@ private:
     std::vector<ActionRequestInfo> _pendingRequests;
     int _numOutstandingActions;
     bool _restartRequested;
+
+    void _requestRestart(OperationContext* opCtx);
 };
 
 class ClusterChunksResizePolicyImpl : public ClusterChunksResizePolicy {