From 0d2f7de3535b9b19520459754788a6953d6727ac Mon Sep 17 00:00:00 2001
From: Rose Jethani <101273941+rose2221@users.noreply.github.com>
Date: Mon, 19 Aug 2024 15:09:15 +0530
Subject: [PATCH] Update NoiseProtocolTests.cs

---
 src/libp2p/Libp2p.Protocols.Noise.Tests/NoiseProtocolTests.cs | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/libp2p/Libp2p.Protocols.Noise.Tests/NoiseProtocolTests.cs b/src/libp2p/Libp2p.Protocols.Noise.Tests/NoiseProtocolTests.cs
index fce5ca91..729aa581 100644
--- a/src/libp2p/Libp2p.Protocols.Noise.Tests/NoiseProtocolTests.cs
+++ b/src/libp2p/Libp2p.Protocols.Noise.Tests/NoiseProtocolTests.cs
@@ -70,8 +70,7 @@ public async Task Test_ConnectionEstablished_With_PreSelectedMuxer()
         //Act
         Task ListenTask = proto.ListenAsync(downChannel, channelFactory, listenerContext);
         Task DialTask = proto.DialAsync(downChannelFromProtocolPov, channelFactory, peerContext);
-        await DialTask;
-        await ListenTask;
+        await Task.WhenAll(DialTask, ListenTask)
 
         //Assert
         Assert.That(peerContext.SpecificProtocolRequest.SubProtocol, Is.EqualTo(proto1));