u32 ieee80211_calc_expected_tx_airtime(struct ieee80211_hw *hw,
                                       struct ieee80211_vif *vif,
                                       struct ieee80211_sta *pubsta,
-                                      int len)
+                                      int len, bool ampdu)
 {
        struct ieee80211_supported_band *sband;
        struct ieee80211_chanctx_conf *conf;
        if (pubsta) {
                struct sta_info *sta = container_of(pubsta, struct sta_info,
                                                    sta);
-
-               return ieee80211_calc_tx_airtime_rate(hw,
-                                                     &sta->tx_stats.last_rate,
-                                                     band, len);
+               struct ieee80211_tx_rate *rate = &sta->tx_stats.last_rate;
+               u32 airtime;
+
+               if (!(rate->flags & (IEEE80211_TX_RC_VHT_MCS |
+                                    IEEE80211_TX_RC_MCS)))
+                       ampdu = false;
+
+               /*
+                * Assume that HT/VHT transmission on any AC except VO will
+                * use aggregation. Since we don't have reliable reporting
+                * of aggregation length, assume an average of 16.
+                * This will not be very accurate, but much better than simply
+                * assuming un-aggregated tx.
+                */
+               airtime = ieee80211_calc_tx_airtime_rate(hw, rate, band,
+                                                        ampdu ? len * 16 : len);
+               if (ampdu)
+                       airtime /= 16;
+
+               return airtime;
        }
 
        if (!conf)
 
 u32 ieee80211_calc_expected_tx_airtime(struct ieee80211_hw *hw,
                                       struct ieee80211_vif *vif,
                                       struct ieee80211_sta *pubsta,
-                                      int len);
+                                      int len, bool ampdu);
 #ifdef CONFIG_MAC80211_NOINLINE
 #define debug_noinline noinline
 #else
 
 
        if (vif &&
            wiphy_ext_feature_isset(local->hw.wiphy, NL80211_EXT_FEATURE_AQL)) {
+               bool ampdu = txq->ac != IEEE80211_AC_VO;
                u32 airtime;
 
                airtime = ieee80211_calc_expected_tx_airtime(hw, vif, txq->sta,
-                                                            skb->len);
+                                                            skb->len, ampdu);
                if (airtime) {
                        airtime = ieee80211_info_set_tx_time_est(info, airtime);
                        ieee80211_sta_update_pending_airtime(local, tx.sta,