static const struct ieee80211_sband_iftype_data iwl_he_eht_capa[] = {
        {
-               .types_mask = BIT(NL80211_IFTYPE_STATION),
+               .types_mask = BIT(NL80211_IFTYPE_STATION) |
+                             BIT(NL80211_IFTYPE_P2P_CLIENT),
                .he_cap = {
                        .has_he = true,
                        .he_cap_elem = {
                },
        },
        {
-               .types_mask = BIT(NL80211_IFTYPE_AP),
+               .types_mask = BIT(NL80211_IFTYPE_AP) |
+                             BIT(NL80211_IFTYPE_P2P_GO),
                .he_cap = {
                        .has_he = true,
                        .he_cap_elem = {
                         u8 tx_chains, u8 rx_chains,
                         const struct iwl_fw *fw)
 {
-       bool is_ap = iftype_data->types_mask & BIT(NL80211_IFTYPE_AP);
+       bool is_ap = iftype_data->types_mask & (BIT(NL80211_IFTYPE_AP) |
+                                               BIT(NL80211_IFTYPE_P2P_GO));
        bool no_320;
 
        no_320 = (!trans->trans_cfg->integrated &&