diff --git a/bin/pwnagotchi b/bin/pwnagotchi
index 234cb374..f6cc89b8 100755
--- a/bin/pwnagotchi
+++ b/bin/pwnagotchi
@@ -323,7 +323,7 @@ def pwnagotchi_cli():
 
     def usr1_handler(*unused):
         logging.info('Received USR1 singal. Restart process ...')
-        restart("MANU" if args.do_manual else "AUTO")
+        agent._restart("MANU" if args.do_manual else "AUTO")
 
     signal.signal(signal.SIGUSR1, usr1_handler)
 
diff --git a/pwnagotchi/agent.py b/pwnagotchi/agent.py
index 9469fd0e..33384923 100644
--- a/pwnagotchi/agent.py
+++ b/pwnagotchi/agent.py
@@ -270,9 +270,9 @@ class Agent(Client, Automata, AsyncAdvertiser, AsyncTrainer):
         self._save_recovery_data()
         pwnagotchi.reboot()
 
-    def _restart(self):
+    def _restart(self, mode='AUTO'):
         self._save_recovery_data()
-        pwnagotchi.restart("AUTO")
+        pwnagotchi.restart(mode)
 
     def _save_recovery_data(self):
         logging.warning("writing recovery data to %s ...", RECOVERY_DATA_FILE)