1 files changed, 2 insertions, 2 deletions
diff --git a/webui.py b/webui.py
index 98b6d394..a0d5e2cd 100644
--- a/webui.py
+++ b/webui.py
@@ -264,13 +264,13 @@ def create_api(app):
 def wait_on_server(demo=None):
     while 1:
         time.sleep(0.5)
-        modules.script_callbacks.app_polling_callback(None, demo)
         if shared.state.need_restart:
-            modules.script_callbacks.app_reload_callback(None, demo)
             shared.state.need_restart = False
             time.sleep(0.5)
             demo.close()
             time.sleep(0.5)
+
+            modules.script_callbacks.app_reload_callback()
             break