local->scan_ies_len +=
                        2 + sizeof(struct ieee80211_vht_cap);
 
-       /* HE cap element is variable in size - set len to allow max size */
        /*
-        * TODO: 1 is added at the end of the calculation to accommodate for
-        *      the temporary placing of the HE capabilities IE under EXT.
-        *      Remove it once it is placed in the final place.
-        */
-       if (supp_he)
+        * HE cap element is variable in size - set len to allow max size */
+       if (supp_he) {
                local->scan_ies_len +=
-                       2 + sizeof(struct ieee80211_he_cap_elem) +
+                       3 + sizeof(struct ieee80211_he_cap_elem) +
                        sizeof(struct ieee80211_he_mcs_nss_supp) +
-                       IEEE80211_HE_PPE_THRES_MAX_LEN + 1;
+                       IEEE80211_HE_PPE_THRES_MAX_LEN;
+       }
 
        if (!local->ops->hw_scan) {
                /* For hw_scan, driver needs to set these up. */
 
        if (!he_cap || !reg_cap)
                return;
 
-       /*
-        * TODO: the 1 added is because this temporarily is under the EXTENSION
-        * IE. Get rid of it when it moves.
-        */
        he_cap_size =
                2 + 1 + sizeof(he_cap->he_cap_elem) +
                ieee80211_he_mcs_nss_size(&he_cap->he_cap_elem) +