diff --git a/scripts/factory_restart_space.py b/scripts/factory_restart_space.py
index 8a52dc42a..07a25d0de 100644
--- a/scripts/factory_restart_space.py
+++ b/scripts/factory_restart_space.py
@@ -18,7 +18,7 @@ if not space:
     print("Please provide a space to restart.")
     exit()
 
-if not parsed_args.api_token:
+if not parsed_args.token:
     print("Please provide an API token.")
     exit()