IEEE80211_EHT_PHY_CAP0_SU_BEAMFORMER |
                IEEE80211_EHT_PHY_CAP0_SU_BEAMFORMEE;
 
+       val = max_t(u8, sts - 1, 3);
        eht_cap_elem->phy_cap_info[0] |=
-               u8_encode_bits(u8_get_bits(sts - 1, BIT(0)),
+               u8_encode_bits(u8_get_bits(val, BIT(0)),
                               IEEE80211_EHT_PHY_CAP0_BEAMFORMEE_SS_80MHZ_MASK);
 
        eht_cap_elem->phy_cap_info[1] =
-               u8_encode_bits(u8_get_bits(sts - 1, GENMASK(2, 1)),
+               u8_encode_bits(u8_get_bits(val, GENMASK(2, 1)),
                               IEEE80211_EHT_PHY_CAP1_BEAMFORMEE_SS_80MHZ_MASK) |
-               u8_encode_bits(sts - 1,
+               u8_encode_bits(val,
                               IEEE80211_EHT_PHY_CAP1_BEAMFORMEE_SS_160MHZ_MASK) |
-               u8_encode_bits(sts - 1,
+               u8_encode_bits(val,
                               IEEE80211_EHT_PHY_CAP1_BEAMFORMEE_SS_320MHZ_MASK);
 
        eht_cap_elem->phy_cap_info[2] =