drm_WARN_ON(&i915->drm, pll->on);
 
        drm_dbg_kms(&i915->drm, "enabling %s\n", pll->info->name);
+
+       if (pll->info->power_domain)
+               pll->wakeref = intel_display_power_get(i915, pll->info->power_domain);
+
        pll->info->funcs->enable(i915, pll);
        pll->on = true;
 
        pll->info->funcs->disable(i915, pll);
        pll->on = false;
 
+       if (pll->info->power_domain)
+               intel_display_power_put(i915, pll->info->power_domain, pll->wakeref);
+
 out:
        mutex_unlock(&i915->display.dpll.lock);
 }
 {
        i915_reg_t enable_reg = intel_combo_pll_enable_reg(i915, pll);
 
-       if (pll->info->power_domain)
-               pll->wakeref = intel_display_power_get(i915, pll->info->power_domain);
-
        icl_pll_power_enable(i915, pll, enable_reg);
 
        icl_dpll_write(i915, pll);
        i915_reg_t enable_reg = intel_combo_pll_enable_reg(i915, pll);
 
        icl_pll_disable(i915, pll, enable_reg);
-
-       if (pll->info->power_domain)
-               intel_display_power_put(i915, pll->info->power_domain, pll->wakeref);
 }
 
 static void tbt_pll_disable(struct drm_i915_private *i915,
 
        pll->info->funcs->disable(i915, pll);
        pll->on = false;
+
+       if (pll->info->power_domain)
+               intel_display_power_put(i915, pll->info->power_domain, pll->wakeref);
 }
 
 void intel_dpll_sanitize_state(struct drm_i915_private *i915)