diff --git a/src/charm.py b/src/charm.py
index 05d8729bb4..351609aad4 100755
--- a/src/charm.py
+++ b/src/charm.py
@@ -1410,7 +1410,7 @@ def _set_primary_status_message(self) -> None:
         """Display 'Primary' in the unit status message if the current unit is the primary."""
         try:
             if "require-change-bucket-after-restore" in self.app_peer_data:
-                if self.charm.unit.is_leader():
+                if self.unit.is_leader():
                     self.app_peer_data.update({
                         "restoring-backup": "",
                         "restore-stanza": "",