diff --git a/actix-server/tests/test_server.rs b/actix-server/tests/test_server.rs
index 6dacc731..920ace51 100644
--- a/actix-server/tests/test_server.rs
+++ b/actix-server/tests/test_server.rs
@@ -475,7 +475,7 @@ async fn test_worker_restart() {
             let mut stream = stream.into_std().unwrap();
             use std::io::Write;
             let str = counter.to_string();
-            stream.write(str.as_bytes()).unwrap();
+            stream.write_all(str.as_bytes()).unwrap();
 
             stream.flush().unwrap();
 
@@ -535,8 +535,6 @@ async fn test_worker_restart() {
     assert_eq!("1", id);
     stream.shutdown().await.unwrap();
 
-    sleep(Duration::from_secs(3)).await;
-
     // worker 2 restarting and work goes to worker 1.
     let mut stream = TcpStream::connect(addr).await.unwrap();
     let n = stream.read(&mut buf).await.unwrap();