diff --git a/tuic-client/src/connection.rs b/tuic-client/src/connection.rs
index 50d72af..16296ad 100644
--- a/tuic-client/src/connection.rs
+++ b/tuic-client/src/connection.rs
@@ -393,7 +393,9 @@ impl Connection {
         loop {
             time::sleep(heartbeat).await;
 
-            if self.is_closed() {
+            let task_count = self.model.task_connect_count() + self.model.task_associate_count();
+
+            if self.is_closed() || task_count == 0 {
                 break;
             }