if (he_cap->has_he)
                        mode |= PHY_TYPE_BIT_HE;
-       } else if (band == NL80211_BAND_5GHZ) {
+       } else if (band == NL80211_BAND_5GHZ || band == NL80211_BAND_6GHZ) {
                mode |= PHY_TYPE_BIT_OFDM;
 
                if (ht_cap->ht_supported)
 
                if (he_cap->has_he)
                        mode |= PHY_MODE_AX_24G;
-       } else if (band == NL80211_BAND_5GHZ) {
+       } else if (band == NL80211_BAND_5GHZ || band == NL80211_BAND_6GHZ) {
                mode |= PHY_MODE_A;
 
                if (ht_cap->ht_supported)
                if (vht_cap->vht_supported)
                        mode |= PHY_MODE_AC;
 
-               if (he_cap->has_he)
-                       mode |= PHY_MODE_AX_5G;
+               if (he_cap->has_he) {
+                       if (band == NL80211_BAND_6GHZ)
+                               mode |= PHY_MODE_AX_6G;
+                       else
+                               mode |= PHY_MODE_AX_5G;
+               }
        }
 
        return mode;