1 files changed, 8 insertions, 5 deletions

diff --git a/drivers/net/wireless/intel/iwlwifi/mvm/d3.c b/drivers/net/wireless/intel/iwlwifi/mvm/d3.c
index b88e2048ae0b..207d8ae1e116 100644
--- a/drivers/net/wireless/intel/iwlwifi/mvm/d3.c
+++ b/drivers/net/wireless/intel/iwlwifi/mvm/d3.c
@@ -1262,12 +1262,15 @@ static int __iwl_mvm_suspend(struct ieee80211_hw *hw,
         iwl_trans_d3_suspend(mvm->trans, test, !unified_image);
  out:
         if (ret < 0) {
-                iwl_mvm_ref(mvm, IWL_MVM_REF_UCODE_DOWN);
-                if (mvm->restart_fw > 0) {
-                        mvm->restart_fw--;
-                        ieee80211_restart_hw(mvm->hw);
-                }
                 iwl_mvm_free_nd(mvm);
+
+                if (!unified_image) {
+                        iwl_mvm_ref(mvm, IWL_MVM_REF_UCODE_DOWN);
+                        if (mvm->restart_fw > 0) {
+                                mvm->restart_fw--;
+                                ieee80211_restart_hw(mvm->hw);
+                        }
+                }
         }
  out_noreset:
         mutex_unlock(&mvm->mutex);