diff --git a/rust/benches/direct_data.rs b/rust/benches/direct_data.rs
index 8406a7acd3..0cea4d8fe0 100644
--- a/rust/benches/direct_data.rs
+++ b/rust/benches/direct_data.rs
@@ -19,7 +19,7 @@ async fn create_data_producer_consumer_pair(
         .produce_data(DataProducerOptions::new_direct())
         .await?;
     let data_consumer = direct_transport
-        .consume_data(DataConsumerOptions::new_direct(data_producer.id()))
+        .consume_data(DataConsumerOptions::new_direct(data_producer.id(), None))
         .await?;
 
     Ok((data_producer, data_consumer))
@@ -51,7 +51,8 @@ pub fn criterion_benchmark(c: &mut Criterion) {
                     let _ = sender.send(());
                 });
 
-                let _ = direct_data_producer.send(WebRtcMessage::Binary(Cow::from(data)));
+                let _ =
+                    direct_data_producer.send(WebRtcMessage::Binary(Cow::from(data)), None, None);
 
                 let _ = receiver.recv();
             })
diff --git a/rust/tests/integration/direct_transport.rs b/rust/tests/integration/direct_transport.rs
index 07f54cad83..380f3e1b3e 100644
--- a/rust/tests/integration/direct_transport.rs
+++ b/rust/tests/integration/direct_transport.rs
@@ -537,14 +537,14 @@ fn send_with_subchannels_succeeds() {
             .expect("Failed tor receive all messages");
 
         let received_messages_1 = received_messages_1.lock().clone();
-        assert_eq!(received_messages_1.contains(&both.to_string()), true);
-        assert_eq!(received_messages_1.contains(&dc1.to_string()), true);
-        assert_eq!(received_messages_1.contains(&dc2.to_string()), false);
+        assert!(received_messages_1.contains(&both.to_string()));
+        assert!(received_messages_1.contains(&dc1.to_string()));
+        assert!(!received_messages_1.contains(&dc2.to_string()));
 
         let received_messages_2 = received_messages_2.lock().clone();
-        assert_eq!(received_messages_2.contains(&both.to_string()), true);
-        assert_eq!(received_messages_2.contains(&dc1.to_string()), false);
-        assert_eq!(received_messages_2.contains(&dc2.to_string()), true);
+        assert!(received_messages_2.contains(&both.to_string()));
+        assert!(!received_messages_2.contains(&dc1.to_string()));
+        assert!(received_messages_2.contains(&dc2.to_string()));
     });
 }