diff --git a/substrate/core/network/src/test/mod.rs b/substrate/core/network/src/test/mod.rs
index f6c866d76d..b6a6103071 100644
--- a/substrate/core/network/src/test/mod.rs
+++ b/substrate/core/network/src/test/mod.rs
@@ -28,7 +28,7 @@ use crate::config::build_multiaddr;
 use log::trace;
 use crate::chain::FinalityProofProvider;
 use client::{
-	self, ClientInfo, BlockchainEvents, BlockImportNotification, FinalityNotifications,
+	self, ClientInfo, BlockchainEvents, BlockImportNotification, FinalityNotifications, ImportNotifications,
 	FinalityNotification, LongestChain
 };
 use client::error::Result as ClientResult;
@@ -189,6 +189,13 @@ impl PeersClient {
 		}
 	}
 
+	pub fn import_notification_stream(&self) -> ImportNotifications<Block>{
+		match *self {
+			PeersClient::Full(ref client, ref _backend) => client.import_notification_stream(),
+			PeersClient::Light(ref client, ref _backend) => client.import_notification_stream(),
+		}
+	}
+
 	pub fn finalize_block(
 		&self,
 		id: BlockId<Block>,