diff --git a/py-scripts/lf_wifi_capacity_test.py b/py-scripts/lf_wifi_capacity_test.py
index 6598cce9..2f9ab4c2 100755
--- a/py-scripts/lf_wifi_capacity_test.py
+++ b/py-scripts/lf_wifi_capacity_test.py
@@ -326,6 +326,7 @@ class WiFiCapacityTest(cv_test):
     def __init__(self,
                  lfclient_host="localhost",
                  lf_port=8080,
+                 ssh_port=22,
                  lf_user="lanforge",
                  lf_password="lanforge",
                  instance_name="wct_instance",
@@ -384,6 +385,7 @@ class WiFiCapacityTest(cv_test):
         self.security = security
         self.ssid = ssid
         self.paswd = paswd
+        self.ssh_port = ssh_port
         self.radio = radio
         self.enables = enables
         self.disables = disables
@@ -476,7 +478,7 @@ class WiFiCapacityTest(cv_test):
         self.create_and_run_test(self.load_old_cfg, self.test_name, self.instance_name,
                                  self.config_name, self.sets,
                                  self.pull_report, self.lfclient_host, self.lf_user, self.lf_password,
-                                 cv_cmds, graph_groups_file=self.graph_groups, local_lf_report_dir=self.local_lf_report_dir)
+                                 cv_cmds, ssh_port=self.ssh_port, graph_groups_file=self.graph_groups, local_lf_report_dir=self.local_lf_report_dir)
 
         self.rm_text_blob(self.config_name, blob_test)  # To delete old config with same name