return ktime_us_delta(now, tx_priv->xmit_timestamp);
 }
 
+bool wfx_tx_queues_has_cab(struct wfx_vif *wvif)
+{
+       struct wfx_dev *wdev = wvif->wdev;
+       struct ieee80211_tx_info *tx_info;
+       struct hif_msg *hif;
+       struct sk_buff *skb;
+       int i;
+
+       for (i = 0; i < IEEE80211_NUM_ACS; ++i) {
+               skb_queue_walk(&wdev->tx_queue[i].queue, skb) {
+                       tx_info = IEEE80211_SKB_CB(skb);
+                       hif = (struct hif_msg *)skb->data;
+                       if ((tx_info->flags & IEEE80211_TX_CTL_SEND_AFTER_DTIM) &&
+                           (hif->interface == wvif->id))
+                               return true;
+               }
+       }
+       return false;
+}
+
 bool wfx_tx_queues_empty(struct wfx_dev *wdev)
 {
        int i;
        return &wvif->wdev->tx_queue[winner];
 }
 
-struct hif_msg *wfx_tx_queues_get_after_dtim(struct wfx_vif *wvif)
-{
-       struct wfx_dev *wdev = wvif->wdev;
-       struct ieee80211_tx_info *tx_info;
-       struct hif_msg *hif;
-       struct sk_buff *skb;
-       int i;
-
-       for (i = 0; i < IEEE80211_NUM_ACS; ++i) {
-               skb_queue_walk(&wdev->tx_queue[i].queue, skb) {
-                       tx_info = IEEE80211_SKB_CB(skb);
-                       hif = (struct hif_msg *)skb->data;
-                       if ((tx_info->flags & IEEE80211_TX_CTL_SEND_AFTER_DTIM) &&
-                           (hif->interface == wvif->id))
-                               return (struct hif_msg *)skb->data;
-               }
-       }
-       return NULL;
-}
-
 struct hif_msg *wfx_tx_queues_get(struct wfx_dev *wdev)
 {
        struct sk_buff *skb;
 
 void wfx_tx_queues_deinit(struct wfx_dev *wdev);
 void wfx_tx_queues_clear(struct wfx_dev *wdev);
 bool wfx_tx_queues_empty(struct wfx_dev *wdev);
+bool wfx_tx_queues_has_cab(struct wfx_vif *wvif);
 void wfx_tx_queues_wait_empty_vif(struct wfx_vif *wvif);
 struct hif_msg *wfx_tx_queues_get(struct wfx_dev *wdev);
-struct hif_msg *wfx_tx_queues_get_after_dtim(struct wfx_vif *wvif);
 
 void wfx_tx_queue_put(struct wfx_dev *wdev, struct wfx_queue *queue,
                      struct sk_buff *skb);
 
                tim_ptr[2] = 0;
 
                /* Set/reset aid0 bit */
-               if (wfx_tx_queues_get_after_dtim(wvif))
+               if (wfx_tx_queues_has_cab(wvif))
                        tim_ptr[4] |= 1;
                else
                        tim_ptr[4] &= ~1;
 
 void wfx_suspend_resume_mc(struct wfx_vif *wvif, enum sta_notify_cmd notify_cmd)
 {
-       WARN(!wfx_tx_queues_get_after_dtim(wvif), "incorrect sequence");
+       WARN(!wfx_tx_queues_has_cab(wvif), "incorrect sequence");
        WARN(wvif->after_dtim_tx_allowed, "incorrect sequence");
        wvif->after_dtim_tx_allowed = true;
        wfx_bh_request_tx(wvif->wdev);