diff --git a/tokio-tcp/src/stream.rs b/tokio-tcp/src/stream.rs
index a8e67c0ae..4f39feab5 100644
--- a/tokio-tcp/src/stream.rs
+++ b/tokio-tcp/src/stream.rs
@@ -618,7 +618,7 @@ impl<'a> futures2::io::AsyncWrite for &'a TcpStream {
                 Ok(futures2::Async::Ready(n))
             }
             Err(ref e) if e.kind() == io::ErrorKind::WouldBlock => {
-                self.io.clear_write_ready()?;
+                self.io.clear_write_ready2(cx)?;
                 Ok(futures2::Async::Pending)
             }
             Err(e) => Err(e),