diff --git a/tests/hwsim/test_ap_wps.py b/tests/hwsim/test_ap_wps.py
index df408f134..e0791a0ca 100644
--- a/tests/hwsim/test_ap_wps.py
+++ b/tests/hwsim/test_ap_wps.py
@@ -10088,6 +10088,7 @@ def run_ap_wps_and_sae(dev, apdev):
     hapd.request("WPS_PIN any " + pin)
 
     dev[0].set("wps_cred_add_sae", "1")
+    dev[0].request("SET sae_groups ")
     dev[0].scan_for_bss(apdev[0]['bssid'], freq="2412", force_scan=True)
     dev[0].request("WPS_PIN " + apdev[0]['bssid'] + " " + pin)
     dev[0].wait_connected(timeout=30)
@@ -10125,6 +10126,7 @@ def run_ap_wps_conf_and_sae(dev, apdev):
                            "rsn_pairwise": "CCMP"})
 
     dev[0].set("wps_cred_add_sae", "1")
+    dev[0].request("SET sae_groups ")
     dev[0].scan_for_bss(apdev[0]['bssid'], freq="2412")
     pin = dev[0].wps_read_pin()
     hapd.request("WPS_PIN any " + pin)
@@ -10154,6 +10156,7 @@ def run_ap_wps_reg_config_and_sae(dev, apdev):
                     "ap_pin": appin, "wps_cred_add_sae": "1"})
     logger.info("WPS configuration step")
     dev[0].set("wps_cred_add_sae", "1")
+    dev[0].request("SET sae_groups ")
     dev[0].scan_for_bss(apdev[0]['bssid'], freq=2412)
     dev[0].dump_monitor()
     new_ssid = "wps-new-ssid"