return cpu_to_le16(bcap->he_6ghz_capa);
 }
 
+static void ath11k_mac_set_hemcsmap(struct ath11k *ar,
+                                   struct ath11k_pdev_cap *cap,
+                                   struct ieee80211_sta_he_cap *he_cap,
+                                   int band)
+{
+       struct ath11k_band_cap *band_cap = &cap->band[band];
+
+       he_cap->he_mcs_nss_supp.rx_mcs_80 =
+               cpu_to_le16(band_cap->he_mcs & 0xffff);
+       he_cap->he_mcs_nss_supp.tx_mcs_80 =
+               cpu_to_le16(band_cap->he_mcs & 0xffff);
+       he_cap->he_mcs_nss_supp.rx_mcs_160 =
+               cpu_to_le16((band_cap->he_mcs >> 16) & 0xffff);
+       he_cap->he_mcs_nss_supp.tx_mcs_160 =
+               cpu_to_le16((band_cap->he_mcs >> 16) & 0xffff);
+       he_cap->he_mcs_nss_supp.rx_mcs_80p80 =
+               cpu_to_le16((band_cap->he_mcs >> 16) & 0xffff);
+       he_cap->he_mcs_nss_supp.tx_mcs_80p80 =
+               cpu_to_le16((band_cap->he_mcs >> 16) & 0xffff);
+}
+
 static int ath11k_mac_copy_he_cap(struct ath11k *ar,
                                  struct ath11k_pdev_cap *cap,
                                  struct ieee80211_sband_iftype_data *data,
                        break;
                }
 
-               he_cap->he_mcs_nss_supp.rx_mcs_80 =
-                       cpu_to_le16(band_cap->he_mcs & 0xffff);
-               he_cap->he_mcs_nss_supp.tx_mcs_80 =
-                       cpu_to_le16(band_cap->he_mcs & 0xffff);
-               he_cap->he_mcs_nss_supp.rx_mcs_160 =
-                       cpu_to_le16((band_cap->he_mcs >> 16) & 0xffff);
-               he_cap->he_mcs_nss_supp.tx_mcs_160 =
-                       cpu_to_le16((band_cap->he_mcs >> 16) & 0xffff);
-               he_cap->he_mcs_nss_supp.rx_mcs_80p80 =
-                       cpu_to_le16((band_cap->he_mcs >> 16) & 0xffff);
-               he_cap->he_mcs_nss_supp.tx_mcs_80p80 =
-                       cpu_to_le16((band_cap->he_mcs >> 16) & 0xffff);
+               ath11k_mac_set_hemcsmap(ar, cap, he_cap, band);
 
                memset(he_cap->ppe_thres, 0, sizeof(he_cap->ppe_thres));
                if (he_cap_elem->phy_cap_info[6] &