From 4f57f0cdae53c8acfa66729356d1426d1b2cae93 Mon Sep 17 00:00:00 2001
From: Asias He <asias@scylladb.com>
Date: Wed, 21 Oct 2015 14:52:54 +0800
Subject: [PATCH] storage_service: Enable restore_replica_count in remove_node

---
 service/storage_service.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/service/storage_service.cc b/service/storage_service.cc
index 146333ca71..d1bd7d54e3 100644
--- a/service/storage_service.cc
+++ b/service/storage_service.cc
@@ -1536,7 +1536,7 @@ future<> storage_service::remove_node(sstring host_id_string) {
         gossiper.advertise_removing(endpoint, host_id, local_host_id);
 
         // kick off streaming commands
-        // restoreReplicaCount(endpoint, myAddress);
+        restore_replica_count(endpoint, my_address).get();
 
         // wait for ReplicationFinishedVerbHandler to signal we're done
         while (!_replicating_nodes.empty()) {