#ifdef CONFIG_PM
 
+static struct ieee80211_vif *wcn36xx_get_first_assoc_vif(struct wcn36xx *wcn)
+{
+       struct wcn36xx_vif *vif_priv = NULL;
+       struct ieee80211_vif *vif = NULL;
+
+       list_for_each_entry(vif_priv, &wcn->vif_list, list) {
+               if (vif_priv->sta_assoc) {
+                       vif = wcn36xx_priv_to_vif(vif_priv);
+                       break;
+               }
+       }
+       return vif;
+}
+
 static int wcn36xx_suspend(struct ieee80211_hw *hw, struct cfg80211_wowlan *wow)
 {
        struct wcn36xx *wcn = hw->priv;
-       int ret;
+       struct ieee80211_vif *vif = NULL;
+       int ret = 0;
 
        wcn36xx_dbg(WCN36XX_DBG_MAC, "mac suspend\n");
 
        flush_workqueue(wcn->hal_ind_wq);
-       ret = wcn36xx_smd_set_power_params(wcn, true);
+       mutex_lock(&wcn->conf_mutex);
+       vif = wcn36xx_get_first_assoc_vif(wcn);
+       if (vif)
+               ret = wcn36xx_smd_set_power_params(wcn, true);
+       mutex_unlock(&wcn->conf_mutex);
 
        return ret;
 }
 static int wcn36xx_resume(struct ieee80211_hw *hw)
 {
        struct wcn36xx *wcn = hw->priv;
+       struct ieee80211_vif *vif = NULL;
 
        wcn36xx_dbg(WCN36XX_DBG_MAC, "mac resume\n");
 
        flush_workqueue(wcn->hal_ind_wq);
-       wcn36xx_smd_set_power_params(wcn, false);
+       mutex_lock(&wcn->conf_mutex);
+       vif = wcn36xx_get_first_assoc_vif(wcn);
+       if (vif)
+               wcn36xx_smd_set_power_params(wcn, false);
+       mutex_unlock(&wcn->conf_mutex);
+
        return 0;
 }