diff --git a/extensions/api/script.py b/extensions/api/script.py
index e57aac92..12fd9cad 100644
--- a/extensions/api/script.py
+++ b/extensions/api/script.py
@@ -7,7 +7,7 @@ from modules import shared
 
 def setup():
     blocking_api.start_server(shared.args.api_blocking_port, share=shared.args.public_api, tunnel_id=shared.args.public_api_id)
-    if shared.public_api:
+    if shared.args.public_api:
         time.sleep(5)
 
     streaming_api.start_server(shared.args.api_streaming_port, share=shared.args.public_api, tunnel_id=shared.args.public_api_id)