From 4d3f333ec0745a09aa37e8a41629fe42e4976d55 Mon Sep 17 00:00:00 2001
From: Asias He <asias@cloudius-systems.com>
Date: Tue, 11 Aug 2015 14:43:12 +0800
Subject: [PATCH] storage_service: Enable call to remove_endpoint in on_remove

---
 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 f5139456fa..5753db09d7 100644
--- a/service/storage_service.cc
+++ b/service/storage_service.cc
@@ -699,8 +699,8 @@ void storage_service::on_change(inet_address endpoint, application_state state,
 
 void storage_service::on_remove(gms::inet_address endpoint) {
     logger.debug("on_remove endpoint={}", endpoint);
+    _token_metadata.remove_endpoint(endpoint);
 #if 0
-    _token_metadata.removeEndpoint(endpoint);
     PendingRangeCalculatorService.instance.update();
 #endif
 }