diff --git a/client/network/src/service/tests/mod.rs b/client/network/src/service/tests/mod.rs
index 94fbf3208fe96..6f35a19aeb9d3 100644
--- a/client/network/src/service/tests/mod.rs
+++ b/client/network/src/service/tests/mod.rs
@@ -80,9 +80,9 @@ impl TestNetwork {
 		let service = worker.service().clone();
 		let event_stream = service.event_stream("test");
 
-		tokio::spawn(async move {
-			worker.run().await;
-		});
+		tokio::spawn(
+			worker.run()
+		);
 
 		(service, event_stream)
 	}
diff --git a/client/network/test/src/lib.rs b/client/network/test/src/lib.rs
index d680b03b3a4cd..77a83d10aeee8 100644
--- a/client/network/test/src/lib.rs
+++ b/client/network/test/src/lib.rs
@@ -1065,9 +1065,8 @@ where
 			for (i, peer) in peers.iter_mut().enumerate() {
 				trace!(target: "sync", "-- Polling {}: {}", i, peer.id());
 				{
-					let net_poll_future = peer.network.run();
-					pin_mut!(net_poll_future);
-					let _ = net_poll_future.poll(cx);
+					peer.network.next_action().poll(cx);
+
 				}
 				trace!(target: "sync", "-- Polling complete {}: {}", i, peer.id());