diff --git a/testing/integration-tests/src/client/mod.rs b/testing/integration-tests/src/client/mod.rs
index d2178e6181..c2a9f9a10b 100644
--- a/testing/integration-tests/src/client/mod.rs
+++ b/testing/integration-tests/src/client/mod.rs
@@ -501,9 +501,11 @@ async fn chainhead_unstable_body() {
     let extrinsics: Vec<Vec<u8>> = body.block.extrinsics.into_iter().map(|ext| ext.0).collect();
     let expected = format!("0x{}", hex::encode(extrinsics.encode()));
 
-    #[cfg(not(feature = "unstable-light-client"))]
     assert_matches!(event,
+        #[cfg(not(feature = "unstable-light-client"))]
         ChainHeadEvent::Done(done) if done.result == expected
+        #[cfg(feature = "unstable-light-client")]
+        ChainHeadEvent::Done(done) if done.value == expected
     );
 }
 
@@ -559,6 +561,8 @@ async fn chainhead_unstable_storage() {
 
     #[cfg(not(feature = "unstable-light-client"))]
     assert_matches!(event, ChainHeadEvent::<Option<String>>::Done(done) if done.result.is_some());
+    #[cfg(feature = "unstable-light-client")]
+    assert_matches!(event, ChainHeadEvent::<Option<String>>::Done(done) if done.value.is_some());
 }
 
 #[tokio::test]