/* Check if there has been a timeout. */
        spin_lock(&wmi->wmi_lock);
-       if (cmd_id != wmi->last_cmd_id) {
+       if (be16_to_cpu(hdr->seq_no) != wmi->last_seq_id) {
                spin_unlock(&wmi->wmi_lock);
                goto free_skb;
        }
                               enum wmi_cmd_id cmd, u16 len)
 {
        struct wmi_cmd_hdr *hdr;
+       unsigned long flags;
 
        hdr = (struct wmi_cmd_hdr *) skb_push(skb, sizeof(struct wmi_cmd_hdr));
        hdr->command_id = cpu_to_be16(cmd);
        hdr->seq_no = cpu_to_be16(++wmi->tx_seq_id);
 
+       spin_lock_irqsave(&wmi->wmi_lock, flags);
+       wmi->last_seq_id = wmi->tx_seq_id;
+       spin_unlock_irqrestore(&wmi->wmi_lock, flags);
+
        return htc_send_epid(wmi->htc, skb, wmi->ctrl_epid);
 }
 
        struct sk_buff *skb;
        u8 *data;
        int time_left, ret = 0;
-       unsigned long flags;
 
        if (ah->ah_flags & AH_UNPLUGGED)
                return 0;
        wmi->cmd_rsp_buf = rsp_buf;
        wmi->cmd_rsp_len = rsp_len;
 
-       spin_lock_irqsave(&wmi->wmi_lock, flags);
-       wmi->last_cmd_id = cmd_id;
-       spin_unlock_irqrestore(&wmi->wmi_lock, flags);
-
        ret = ath9k_wmi_cmd_issue(wmi, skb, cmd_id, cmd_len);
        if (ret)
                goto out;
 
        enum htc_endpoint_id ctrl_epid;
        struct mutex op_mutex;
        struct completion cmd_wait;
-       enum wmi_cmd_id last_cmd_id;
+       u16 last_seq_id;
        struct sk_buff_head wmi_event_queue;
        struct tasklet_struct wmi_event_tasklet;
        u16 tx_seq_id;