diff --git a/demo/nas/sa_nas_mobilenetv2_cifar10.py b/demo/nas/sa_nas_mobilenetv2_cifar10.py
index 3e903960b1c783c38d672238d5a2b3a0c1581c4d..249d4c214788c0ffc5a0d741dc48b4942ea5808b 100644
--- a/demo/nas/sa_nas_mobilenetv2_cifar10.py
+++ b/demo/nas/sa_nas_mobilenetv2_cifar10.py
@@ -39,7 +39,7 @@ def init_sa_nas(config):
         search_steps = 10000000
 
         ### start a server and a client
-        sa_nas = SANAS(config, max_flops=base_flops, search_steps=search_steps)
+        sa_nas = SANAS(config, search_steps=search_steps, is_server=True)
 
         ### start a client, server_addr is server address
         #sa_nas = SANAS(config, max_flops = base_flops, server_addr=("10.255.125.38", 18607), search_steps = search_steps, is_server=False)