diff --git a/src/client.rs b/src/client.rs
index 9b0eb8c..3c961a4 100644
--- a/src/client.rs
+++ b/src/client.rs
@@ -119,11 +119,13 @@ pub async fn client_mode(client_config: ClientConfiguration, fd: i32, env: &JNIE
 
     let s_cipher = cipher_shared.clone();
     loop {
-        //if close_token.is_cancelled() {
+        if close_token.is_cancelled() {
             //sock_read_task.abort();
             //dev_read_task.abort();
             //return;
-        //}
+            ltx.send("Closed token called".as_bytes().to_vec());
+            return;
+        }
         if let Ok(bytes) = rx.try_recv() {
             ltx.send("Write to tun.".as_bytes().to_vec());
             if let Err(e) = dev_writer.write_all(&bytes).await {