diff --git a/src/client.rs b/src/client.rs
index cf05960..471f0c8 100644
--- a/src/client.rs
+++ b/src/client.rs
@@ -1,6 +1,7 @@
use crossbeam_channel::unbounded;
use socket2::SockAddr;
use tokio::{net::UdpSocket, sync::Mutex, io::{BufReader, BufWriter, AsyncWriteExt, AsyncReadExt}, fs::File};
+use tokio_util::sync::CancellationToken;
use base64::prelude::*;
use log::{error, info, warn};
use std::sync::Arc;
diff --git a/src/main.rs b/src/main.rs
index 27fb113..04a0fda 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -23,7 +23,7 @@ mod client;
mod udp;
mod mobile;
-static TUN_QUIT: tokio::sync::Mutex