diff --git a/cc_module_9800_3504.py b/cc_module_9800_3504.py
index 37b45efe..417901a6 100755
--- a/cc_module_9800_3504.py
+++ b/cc_module_9800_3504.py
@@ -583,9 +583,9 @@ INCLUDE_IN_README
     cs.show_ap_dot11_5gz_shutdown()
     # cs.show_ap_dot11_24gz_shutdown() not in txpower
     # This needs to be here to disable and delete
-    cs.wlan = 'wpa2_wlan_4'
-    cs.wlanID = '4'
-    cs.wlanSSID = 'wpa2_wlan_4'
+    cs.wlan = 'wpa2_wlan_3'
+    cs.wlanID = '3'
+    cs.wlanSSID = 'wpa2_wlan_3'
 
     # disable_wlan
     cs.wlan_shutdown()
@@ -633,8 +633,6 @@ INCLUDE_IN_README
     # TODO (there were two in tx_power the logs)
     # need to check if wlan present
     cs.wlan = 'wpa2_wlan_3'
-    cs.wlanID = '3'
-    cs.wlanSSID = 'wpa2_wlan_3'
 
     # delete wlan
     cs.config_no_wlan()
@@ -657,10 +655,11 @@ INCLUDE_IN_README
     cs.security_key = 'hello123'
     cs.config_wlan_wpa2()
 
-    # # create_wlan_wpa3
+    # create_wlan_wpa3
     # cs.wlan = 'wpa3_wlan_4'
     # cs.wlanID = '4'
     # cs.wlanSSID = 'wpa3_wlan_4'
+    # cs.security_key = 'hello123'
     # cs.config_wlan_wpa3()
 
     # wireless_tag_policy
diff --git a/wifi_ctl_9800_3504.py b/wifi_ctl_9800_3504.py
index 470562a1..f5cb7457 100755
--- a/wifi_ctl_9800_3504.py
+++ b/wifi_ctl_9800_3504.py
@@ -1320,7 +1320,7 @@ def main():
                         "radio policy dot11 5ghz",
                         "security wpa psk set-key ascii 0 {security_key}".format(security_key=args.security_key),
                         "no security wpa akm dot1x",
-                        "security wpa akm psk"
+                        "security wpa akm psk",
                         "no shutdown"]:
                         egg.sendline(command)
                         sleep(1)