1 files changed, 7 insertions, 2 deletions
diff --git a/light-client/src/network.rs b/light-client/src/network.rs
index f2c42518..3eb3f50a 100644
--- a/light-client/src/network.rs
+++ b/light-client/src/network.rs
@@ -59,9 +59,14 @@ impl Network {
         });
         let stream = TcpStream::connect((host, port))?;
         stream.set_nodelay(true).unwrap();
-        // stream.set_nonblocking(true).unwrap();
 
-        let (sock, _) = client_tls_with_config(request, stream, None, None).unwrap();
+        let (mut sock, _) = client_tls_with_config(request, stream, None, None).unwrap();
+
+        match sock.get_mut() {
+            MaybeTlsStream::Plain(s) => s.set_nonblocking(true).unwrap(),
+            MaybeTlsStream::Rustls(s) => s.sock.set_nonblocking(true).unwrap(),
+            _ => todo!(),
+        };
 
         Ok(Self {
             sock,