From 94dc345d8ef5fb95fd855a665b1d32f3b0e32008 Mon Sep 17 00:00:00 2001
From: Ivan Folgueira Bande <ivansete@status.im>
Date: Tue, 17 Sep 2024 14:01:08 +0200
Subject: [PATCH] waku_test.go: adapt TestWakuV2Store

---
 wakuv2/waku_test.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/wakuv2/waku_test.go b/wakuv2/waku_test.go
index 1495d1beb..778344498 100644
--- a/wakuv2/waku_test.go
+++ b/wakuv2/waku_test.go
@@ -560,11 +560,11 @@ func TestWakuV2Store(t *testing.T) {
 	}()
 
 	// Connect the two nodes directly
-	peer2Addr := w2.ListenAddresses()[0].String()
-	err = w1.DialPeer(context.Background(), peer2Addr)
+	peer2Addr, err := w2.ListenAddresses()
 	require.NoError(t, err)
 
-	waitForPeerConnection(t, w2.Host().ID(), w1PeersCh)
+	err = w1.DialPeer(peer2Addr[0])
+	require.NoError(t, err)
 
 	// Create a filter for the second node to catch messages
 	filter := &common.Filter{