diff --git a/src/Ui/UiWebsocket.py b/src/Ui/UiWebsocket.py
index ddbb6839..d8441af6 100644
--- a/src/Ui/UiWebsocket.py
+++ b/src/Ui/UiWebsocket.py
@@ -1148,10 +1148,20 @@ class UiWebsocket(object):
@flag.no_multiuser
def actionServerShutdown(self, to, restart=False):
import main
+ def cbServerShutdown(res):
+ self.response(to, res)
+ if not res:
+ return False
+ if restart:
+ main.restart_after_shutdown = True
+ main.file_server.stop()
+ main.ui_server.stop()
+
if restart:
- main.restart_after_shutdown = True
- main.file_server.stop()
- main.ui_server.stop()
+ message = [_["Restart ZeroNet client?"], _["Restart"]]
+ else:
+ message = [_["Shut down ZeroNet client?"], _["Shut down"]]
+ self.cmd("confirm", message, cbServerShutdown)
@flag.admin
@flag.no_multiuser