chore: unify ota and init states

1 files changed, 3 insertions, 4 deletions

diff --git a/embassy-net-esp-hosted/src/lib.rs b/embassy-net-esp-hosted/src/lib.rs
index f0cf04d04..2c7377281 100644
--- a/embassy-net-esp-hosted/src/lib.rs
+++ b/embassy-net-esp-hosted/src/lib.rs
@@ -234,13 +234,12 @@ where
                     tx_buf[..PayloadHeader::SIZE].fill(0);
                 }
                 Either4::Fourth(()) => {
-                    // Extend the deadline if OTA
-                    if self.shared.is_ota() {
+                    // Extend the deadline if initializing
+                    if let ioctl::ControlState::Reboot = self.shared.state() {
                         self.heartbeat_deadline = Instant::now() + HEARTBEAT_MAX_GAP;
                         continue;
-                    } else {
-                        panic!("heartbeat from esp32 stopped")
                     }
+                    panic!("heartbeat from esp32 stopped")
                 }
             }