From 21501815dd8ad9e316d436b3ab7b2dac28807c41 Mon Sep 17 00:00:00 2001
From: Ethan Frey <ethanfrey@users.noreply.github.com>
Date: Mon, 27 Feb 2017 14:55:29 +0100
Subject: [PATCH] Fix EventSwitch usage in WaitForOneEvent

---
 rpc/client/helpers.go | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/rpc/client/helpers.go b/rpc/client/helpers.go
index d73c4ece8..bd00c1438 100644
--- a/rpc/client/helpers.go
+++ b/rpc/client/helpers.go
@@ -72,9 +72,8 @@ func WaitForOneEvent(evsw types.EventSwitch,
 		evts <- data
 	})
 	// make sure to unregister after the test is over
-	// TODO: why doesn't the other call work???
-	// defer evsw.RemoveListenerForEvent(listener, evtTyp)
-	defer evsw.RemoveListener(listener)
+	defer evsw.RemoveListenerForEvent(evtTyp, listener)
+	// defer evsw.RemoveListener(listener)  // this also works
 
 	select {
 	case <-quit: