diff --git a/src/libp2p/Libp2p.Protocols.Pubsub.Tests/PubsubProtocolTests.cs b/src/libp2p/Libp2p.Protocols.Pubsub.Tests/PubsubProtocolTests.cs
index 2b1d99a3..caf4f49b 100644
--- a/src/libp2p/Libp2p.Protocols.Pubsub.Tests/PubsubProtocolTests.cs
+++ b/src/libp2p/Libp2p.Protocols.Pubsub.Tests/PubsubProtocolTests.cs
@@ -23,6 +23,7 @@ public async Task Test_Peer_is_dialed_when_added_by_discovery()
 
         TestDiscoveryProtocol discovery = new();
         CancellationToken token = default;
+        TaskCompletionSource taskCompletionSource = new();
 
         _ = router.RunAsync(peer, discovery, token: token);
         discovery.OnAddPeer!([discoveredPeer]);
@@ -30,7 +31,8 @@ public async Task Test_Peer_is_dialed_when_added_by_discovery()
         await Task.Delay(100);
         _ = peer.Received().DialAsync(discoveredPeer, Arg.Any<CancellationToken>());
 
-        router.OutboundConnection(discoveredPeer, PubsubRouter.FloodsubProtocolVersion, Task.CompletedTask, (rpc) => { });
+        router.OutboundConnection(discoveredPeer, PubsubRouter.FloodsubProtocolVersion, taskCompletionSource.Task, (rpc) => { });
         Assert.That(state.ConnectedPeers, Has.Member(peerId));
+        taskCompletionSource.SetResult();
     }
 }