From 10d54052d7ed70db5d65101179cc4676f9aa8f08 Mon Sep 17 00:00:00 2001
From: Shayan Patel <shayan.patel@canonical.com>
Date: Thu, 5 Sep 2024 03:58:21 +0000
Subject: [PATCH] Re-add unnecessarily removed code

---
 src/charm.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/charm.py b/src/charm.py
index aea8882ae..ad90f93c6 100755
--- a/src/charm.py
+++ b/src/charm.py
@@ -762,6 +762,9 @@ def _handle_potential_cluster_crash_scenario(self) -> bool:
                 self.peers.data[unit].get("member-state", "unknown") for unit in self.peers.units
             }
 
+            # Add state for this unit (self.peers.units does not include this unit)
+            all_states.add("offline")
+
             total_cluster_node_count = self.total_cluster_node_count
             if (
                 all_states == {"offline"} and self.unit.is_leader()