From 9c3dce7cf42f18fb81ea2514da5dcd98572fbf94 Mon Sep 17 00:00:00 2001
From: Aarsh Shah <aarshkshah1992@gmail.com>
Date: Thu, 2 Apr 2020 12:49:53 +0530
Subject: [PATCH] changes as per review

---
 dht_test.go           | 2 +-
 subscriber_notifee.go | 5 ++---
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/dht_test.go b/dht_test.go
index 9fee50898..5025769e0 100644
--- a/dht_test.go
+++ b/dht_test.go
@@ -1414,7 +1414,7 @@ func testFindPeerQuery(t *testing.T,
 	}
 
 	// give some time for things to settle down
-	time.Sleep(2 * time.Second)
+	waitForWellFormedTables(t, dhts, minRTRefreshThreshold, minRTRefreshThreshold, 5*time.Second)
 
 	for _, d := range dhts {
 		if err := <-d.RefreshRoutingTable(); err != nil {
diff --git a/subscriber_notifee.go b/subscriber_notifee.go
index f2474fe72..c80940221 100644
--- a/subscriber_notifee.go
+++ b/subscriber_notifee.go
@@ -120,11 +120,10 @@ func handlePeerProtocolsUpdatedEvent(dht *IpfsDHT, e event.EvtPeerProtocolsUpdat
 
 	if !valid {
 		dht.peerStoppedDHT(dht.ctx, e.Peer)
-	} else {
-		// we just might have discovered a peer that supports the DHT protocol
-		dht.fixLowPeers()
 	}
 
+	// we just might have discovered a peer that supports the DHT protocol
+	dht.fixLowPeers()
 }
 
 func handleLocalReachabilityChangedEvent(dht *IpfsDHT, e event.EvtLocalReachabilityChanged) {