diff --git a/test-utils/client/src/lib.rs b/test-utils/client/src/lib.rs
index 060d487967583..f64e7e3cfb7e8 100644
--- a/test-utils/client/src/lib.rs
+++ b/test-utils/client/src/lib.rs
@@ -384,9 +384,11 @@ where
 
 		Box::pin(async move {
 			while let Some(notification) = import_notification_stream.next().await {
-				blocks.insert(notification.hash);
-				if blocks.len() == count {
-					break;
+				if notification.is_new_best {
+					blocks.insert(notification.hash);
+					if blocks.len() == count {
+						break;
+					}
 				}
 			}
 		})