diff --git a/pkgs/web_socket_channel/test/io_test.dart b/pkgs/web_socket_channel/test/io_test.dart
index ada0608..f9032cd 100644
--- a/pkgs/web_socket_channel/test/io_test.dart
+++ b/pkgs/web_socket_channel/test/io_test.dart
@@ -75,7 +75,7 @@
     server.transform(WebSocketTransformer()).listen((webSocket) {
       expect(() async {
         var channel = IOWebSocketChannel(webSocket);
-        await channel.stream.listen(null).asFuture();
+        await channel.stream.drain();
         expect(channel.closeCode, equals(5678));
         expect(channel.closeReason, equals("raisin"));
       }(), completes);
@@ -94,7 +94,7 @@
     });
 
     var channel = IOWebSocketChannel.connect("ws://localhost:${server.port}");
-    expect(channel.stream.toList(),
+    expect(channel.stream.drain(),
         throwsA(TypeMatcher<WebSocketChannelException>()));
   });
 
@@ -111,7 +111,7 @@
 
     var channel = IOWebSocketChannel.connect("ws://localhost:${server.port}",
         protocols: [failedProtocol]);
-    expect(channel.stream.toList(),
+    expect(channel.stream.drain(),
         throwsA(TypeMatcher<WebSocketChannelException>()));
   });
 
@@ -129,7 +129,7 @@
 
     var channel = IOWebSocketChannel.connect("ws://localhost:${server.port}",
         protocols: [passedProtocol]);
-    await channel.stream.listen(null).asFuture();
+    await channel.stream.drain();
     expect(channel.protocol, passedProtocol);
   });
 }