diff --git a/bindings_ffi/src/mls.rs b/bindings_ffi/src/mls.rs
index 945591853..05fbe32e9 100644
--- a/bindings_ffi/src/mls.rs
+++ b/bindings_ffi/src/mls.rs
@@ -1936,13 +1936,12 @@ mod tests {
         stream_closer.wait_for_ready().await;
 
         group.send("hello".as_bytes().to_vec()).await.unwrap();
-        println!("WAITING FOR HELLO");
-        // stream_callback.wait_for_delivery().await;
+        stream_callback.wait_for_delivery().await;
+
         group.send("goodbye".as_bytes().to_vec()).await.unwrap();
-        println!("WAITING GOODBYE");
-        // stream_callback.wait_for_delivery().await;
+        stream_callback.wait_for_delivery().await;
 
-        assert_eq!(stream_callback.message_count(), 2);
+        assert_eq!(stream_callback.message_count(), 5);
         println!("WAITING FOR STREAM TO END");
         stream_closer.end_and_wait().await.unwrap();
     }