struct rtw_vif *rtwvif = (struct rtw_vif *)vif->drv_priv;
        u32 config = 0;
 
-       rtw_leave_lps(rtwdev);
-
        mutex_lock(&rtwdev->mutex);
 
+       rtw_leave_lps(rtwdev);
+
        ether_addr_copy(rtwvif->mac_addr, mac_addr);
        config |= PORT_SET_MAC_ADDR;
        rtw_vif_port_config(rtwdev, rtwvif, config);
 
        struct rtw_watch_dog_iter_data data = {};
        bool busy_traffic = test_bit(RTW_FLAG_BUSY_TRAFFIC, rtwdev->flags);
 
+       mutex_lock(&rtwdev->mutex);
+
        if (!test_bit(RTW_FLAG_RUNNING, rtwdev->flags))
-               return;
+               goto unlock;
 
        ieee80211_queue_delayed_work(rtwdev->hw, &rtwdev->watch_dog_work,
                                     RTW_WATCH_DOG_DELAY_TIME);
                rtw_leave_lps(rtwdev);
 
        if (test_bit(RTW_FLAG_SCANNING, rtwdev->flags))
-               return;
+               goto unlock;
 
        rtw_phy_dynamic_mechanism(rtwdev);
 
        rtwdev->watch_dog_cnt++;
+
+unlock:
+       mutex_unlock(&rtwdev->mutex);
 }
 
 static void rtw_c2h_work(struct work_struct *work)
 
 {
        struct rtw_lps_conf *conf = &rtwdev->lps_conf;
 
+       lockdep_assert_held(&rtwdev->mutex);
+
        if (test_bit(RTW_FLAG_LEISURE_PS, rtwdev->flags))
                return;
 
 {
        struct rtw_lps_conf *conf = &rtwdev->lps_conf;
 
+       lockdep_assert_held(&rtwdev->mutex);
+
        if (!test_bit(RTW_FLAG_LEISURE_PS, rtwdev->flags))
                return;