From d14fa5880f7ccf771340727d55516650e6f7fdb1 Mon Sep 17 00:00:00 2001
From: AZ Computer Guru <dev@azcomputerguru.com>
Date: Sun, 28 Dec 2025 12:44:28 -0700
Subject: [PATCH] Fix transport variable reference

---
 agent/src/session/mod.rs | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/agent/src/session/mod.rs b/agent/src/session/mod.rs
index 2339bc2..ce689d0 100644
--- a/agent/src/session/mod.rs
+++ b/agent/src/session/mod.rs
@@ -128,8 +128,13 @@ impl SessionManager {
             tokio::time::sleep(Duration::from_millis(1)).await;
 
             // Check if still connected
-            if !transport.is_connected() {
-                tracing::warn!("Connection lost");
+            if let Some(transport) = self.transport.as_ref() {
+                if !transport.is_connected() {
+                    tracing::warn!("Connection lost");
+                    break;
+                }
+            } else {
+                tracing::warn!("Transport is None");
                 break;
             }
         }