diff --git a/tools/aws_benchmarking/paddle_banchmarking_aws.py b/tools/aws_benchmarking/paddle_banchmarking_aws.py
index f6512d923ce6fb1a81aea7f53514e883bab3f5fd..c63e4b0742e4d5a7108d92c675d54a54ecc640be 100644
--- a/tools/aws_benchmarking/paddle_banchmarking_aws.py
+++ b/tools/aws_benchmarking/paddle_banchmarking_aws.py
@@ -28,6 +28,19 @@ import paramiko
 # ~/.aws/credentials and ~/.aws/config
 
 parser = argparse.ArgumentParser(description=__doc__)
+parser.add_argument(
+    '--key_name',
+    type=str,
+    default="",
+    required=True,
+    help="required, key pair name")
+parser.add_argument(
+    '--security_group_id',
+    type=str,
+    default="",
+    required=True,
+    help="required, the security group id associated with your VPC")
+
 parser.add_argument(
     '--vpc_id',
     type=str,
@@ -38,12 +51,7 @@ parser.add_argument(
     type=str,
     default="",
     help="The Subnet_id in which you wish to run test")
-parser.add_argument(
-    '--security_group_id',
-    type=str,
-    default="",
-    required=True,
-    help="required, the security group id associated with your VPC")
+
 parser.add_argument(
     '--pserver_instance_type',
     type=str,
@@ -54,12 +62,7 @@ parser.add_argument(
     type=str,
     default="p2.xlarge",
     help="your trainer instance type")
-parser.add_argument(
-    '--key_name',
-    type=str,
-    default="",
-    required=True,
-    help="required, key pair name")
+
 parser.add_argument(
     '--task_name',
     type=str,
@@ -316,7 +319,6 @@ def cleanup(task_name):
     }])
 
     ec2client.delete_subnet(SubnetId=subnet["Subnets"][0]["SubnetId"])
-
     # no subnet delete waiter, just leave it.
     return