diff --git a/tokio/tests/uds_stream.rs b/tokio/tests/uds_stream.rs
index 5f1b4cffbcc..b8c4e6a8eed 100644
--- a/tokio/tests/uds_stream.rs
+++ b/tokio/tests/uds_stream.rs
@@ -25,13 +25,13 @@ async fn accept_read_write() -> std::io::Result<()> {
     let connect = UnixStream::connect(&sock_path);
     let ((mut server, _), mut client) = try_join(accept, connect).await?;
 
-    // Write to the client. TODO: Switch to write_all.
-    let write_len = client.write(b"hello").await?;
-    assert_eq!(write_len, 5);
+    // Write to the client.
+    client.write_all(b"hello").await?;
     drop(client);
-    // Read from the server. TODO: Switch to read_to_end.
-    let mut buf = [0u8; 5];
-    server.read_exact(&mut buf).await?;
+
+    // Read from the server.
+    let mut buf = vec![];
+    server.read_to_end(&mut buf).await?;
     assert_eq!(&buf, b"hello");
     let len = server.read(&mut buf).await?;
     assert_eq!(len, 0);