diff --git a/primary/src/header_waiter.rs b/primary/src/header_waiter.rs
index 497603181..094e5f395 100644
--- a/primary/src/header_waiter.rs
+++ b/primary/src/header_waiter.rs
@@ -314,10 +314,8 @@ impl HeaderWaiter {
                     match message {
                         ReconfigureNotification::NewEpoch(new_committee) => {
                             // Update the committee and cleanup internal state.
-                            self.primary_network
-                            .cleanup(self.committee.network_diff(&committee));
-                            self.worker_network
-                                .cleanup(self.committee.network_diff(&committee));
+                            self.primary_network.cleanup(self.committee.network_diff(&new_committee));
+                            self.worker_network.cleanup(self.committee.network_diff(&new_committee));
 
                             self.committee = new_committee;