return count;
 }
 
+static void ufshcd_init_clk_scaling(struct ufs_hba *hba)
+{
+       char wq_name[sizeof("ufs_clkscaling_00")];
+
+       if (!ufshcd_is_clkscaling_supported(hba))
+               return;
+
+       INIT_WORK(&hba->clk_scaling.suspend_work,
+                 ufshcd_clk_scaling_suspend_work);
+       INIT_WORK(&hba->clk_scaling.resume_work,
+                 ufshcd_clk_scaling_resume_work);
+
+       snprintf(wq_name, sizeof(wq_name), "ufs_clkscaling_%d",
+                hba->host->host_no);
+       hba->clk_scaling.workq = create_singlethread_workqueue(wq_name);
+
+       ufshcd_clkscaling_init_sysfs(hba);
+}
+
+static void ufshcd_exit_clk_scaling(struct ufs_hba *hba)
+{
+       if (!ufshcd_is_clkscaling_supported(hba))
+               return;
+
+       destroy_workqueue(hba->clk_scaling.workq);
+       ufshcd_devfreq_remove(hba);
+}
+
 static void ufshcd_init_clk_gating(struct ufs_hba *hba)
 {
        char wq_name[sizeof("ufs_clk_gating_00")];
         */
        if (ret && !ufshcd_eh_in_progress(hba) && !hba->pm_op_in_progress) {
                pm_runtime_put_sync(hba->dev);
+               ufshcd_exit_clk_scaling(hba);
                ufshcd_hba_exit(hba);
        }
 
                ufshcd_variant_hba_exit(hba);
                ufshcd_setup_vreg(hba, false);
                ufshcd_suspend_clkscaling(hba);
-               if (ufshcd_is_clkscaling_supported(hba)) {
+               if (ufshcd_is_clkscaling_supported(hba))
                        if (hba->devfreq)
                                ufshcd_suspend_clkscaling(hba);
-                       destroy_workqueue(hba->clk_scaling.workq);
-                       ufshcd_devfreq_remove(hba);
-               }
                ufshcd_setup_clocks(hba, false);
                ufshcd_setup_hba_vreg(hba, false);
                hba->is_powered = false;
        ufshcd_disable_intr(hba, hba->intr_mask);
        ufshcd_hba_stop(hba, true);
 
+       ufshcd_exit_clk_scaling(hba);
        ufshcd_exit_clk_gating(hba);
        if (ufshcd_is_clkscaling_supported(hba))
                device_remove_file(hba->dev, &hba->clk_scaling.enable_attr);
 
        ufshcd_init_clk_gating(hba);
 
+       ufshcd_init_clk_scaling(hba);
+
        /*
         * In order to avoid any spurious interrupt immediately after
         * registering UFS controller interrupt handler, clear any pending UFS
                goto out_remove_scsi_host;
        }
 
-       if (ufshcd_is_clkscaling_supported(hba)) {
-               char wq_name[sizeof("ufs_clkscaling_00")];
-
-               INIT_WORK(&hba->clk_scaling.suspend_work,
-                         ufshcd_clk_scaling_suspend_work);
-               INIT_WORK(&hba->clk_scaling.resume_work,
-                         ufshcd_clk_scaling_resume_work);
-
-               snprintf(wq_name, sizeof(wq_name), "ufs_clkscaling_%d",
-                        host->host_no);
-               hba->clk_scaling.workq = create_singlethread_workqueue(wq_name);
-
-               ufshcd_clkscaling_init_sysfs(hba);
-       }
-
        /*
         * Set the default power management level for runtime and system PM.
         * Default power saving mode is to keep UFS link in Hibern8 state
 out_remove_scsi_host:
        scsi_remove_host(hba->host);
 exit_gating:
+       ufshcd_exit_clk_scaling(hba);
        ufshcd_exit_clk_gating(hba);
 out_disable:
        hba->is_irq_enabled = false;