mgmt->u.action.u.addba_resp.action_code = WLAN_ACTION_ADDBA_RESP;
        mgmt->u.action.u.addba_resp.dialog_token = dialog_token;
 
-       capab = (u16)(amsdu << 0);      /* bit 0 A-MSDU support */
-       capab |= (u16)(policy << 1);    /* bit 1 aggregation policy */
-       capab |= (u16)(tid << 2);       /* bit 5:2 TID number */
-       capab |= (u16)(buf_size << 6);  /* bit 15:6 max size of aggregation */
+       capab = u16_encode_bits(amsdu, IEEE80211_ADDBA_PARAM_AMSDU_MASK);
+       capab |= u16_encode_bits(policy, IEEE80211_ADDBA_PARAM_POLICY_MASK);
+       capab |= u16_encode_bits(tid, IEEE80211_ADDBA_PARAM_TID_MASK);
+       capab |= u16_encode_bits(buf_size, IEEE80211_ADDBA_PARAM_BUF_SIZE_MASK);
 
        mgmt->u.action.u.addba_resp.capab = cpu_to_le16(capab);
        mgmt->u.action.u.addba_resp.timeout = cpu_to_le16(timeout);
 
        mgmt->u.action.u.addba_req.action_code = WLAN_ACTION_ADDBA_REQ;
 
        mgmt->u.action.u.addba_req.dialog_token = dialog_token;
-       capab = (u16)(1 << 0);          /* bit 0 A-MSDU support */
-       capab |= (u16)(1 << 1);         /* bit 1 aggregation policy */
-       capab |= (u16)(tid << 2);       /* bit 5:2 TID number */
-       capab |= (u16)(agg_size << 6);  /* bit 15:6 max size of aggergation */
+       capab = IEEE80211_ADDBA_PARAM_AMSDU_MASK;
+       capab |= IEEE80211_ADDBA_PARAM_POLICY_MASK;
+       capab |= u16_encode_bits(tid, IEEE80211_ADDBA_PARAM_TID_MASK);
+       capab |= u16_encode_bits(agg_size, IEEE80211_ADDBA_PARAM_BUF_SIZE_MASK);
 
        mgmt->u.action.u.addba_req.capab = cpu_to_le16(capab);
 
 
        capab = le16_to_cpu(mgmt->u.action.u.addba_resp.capab);
        amsdu = capab & IEEE80211_ADDBA_PARAM_AMSDU_MASK;
-       tid = (capab & IEEE80211_ADDBA_PARAM_TID_MASK) >> 2;
-       buf_size = (capab & IEEE80211_ADDBA_PARAM_BUF_SIZE_MASK) >> 6;
+       tid = u16_get_bits(capab, IEEE80211_ADDBA_PARAM_TID_MASK);
+       buf_size = u16_get_bits(capab, IEEE80211_ADDBA_PARAM_BUF_SIZE_MASK);
        buf_size = min(buf_size, local->hw.max_tx_aggregation_subframes);
 
        txq = sta->sta.txq[tid];