diff --git a/remoting/zookeeper/listener_test.go b/remoting/zookeeper/listener_test.go
index b1f0d43d3429a38d8d5e9b5ad804dfecd2ee53c1..b02bcbf693b1d05a0591939250a510c698958282 100644
--- a/remoting/zookeeper/listener_test.go
+++ b/remoting/zookeeper/listener_test.go
@@ -111,7 +111,7 @@ func (m *mockDataListener) DataChange(eventType remoting.Event) bool {
 	fmt.Println(eventType)
 	m.eventList = append(m.eventList, eventType)
 	if eventType.Content == m.changedData {
-		m.client.Wait.Done()
+		m.client.Close()
 	}
 	return true
 }