diff --git a/wpa_supplicant/sme.c b/wpa_supplicant/sme.c
index af761243b..ecddb3576 100644
--- a/wpa_supplicant/sme.c
+++ b/wpa_supplicant/sme.c
@@ -1136,8 +1136,7 @@ static int sme_handle_external_auth_start(struct wpa_supplicant *wpa_s,
 		if (!wpas_network_disabled(wpa_s, ssid) &&
 		    ssid_str_len == ssid->ssid_len &&
 		    os_memcmp(ssid_str, ssid->ssid, ssid_str_len) == 0 &&
-		    (ssid->key_mgmt & (WPA_KEY_MGMT_SAE |
-				       WPA_KEY_MGMT_FT_SAE))) {
+		    wpa_key_mgmt_sae(ssid->key_mgmt)) {
 			/* Make sure PT is derived */
 			wpa_s_setup_sae_pt(wpa_s->conf, ssid);
 			break;