return ret;
 }
 
-static int ufs_mtk_unipro_set_pm(struct ufs_hba *hba, u32 lpm)
+static int ufs_mtk_unipro_set_pm(struct ufs_hba *hba, bool lpm)
 {
        int ret;
        struct ufs_mtk_host *host = ufshcd_get_variant(hba);
        ret = ufshcd_dme_set(hba,
                             UIC_ARG_MIB_SEL(VS_UNIPROPOWERDOWNCONTROL, 0),
                             lpm);
-       if (!ret)
+       if (!ret || !lpm) {
+               /*
+                * Forcibly set as non-LPM mode if UIC commands is failed
+                * to use default hba_enable_delay_us value for re-enabling
+                * the host.
+                */
                host->unipro_lpm = lpm;
+       }
 
        return ret;
 }
        int ret;
        u32 tmp;
 
-       ufs_mtk_unipro_set_pm(hba, 0);
+       ret = ufs_mtk_unipro_set_pm(hba, false);
+       if (ret)
+               return ret;
 
        /*
         * Setting PA_Local_TX_LCC_Enable to 0 before link startup
        if (err)
                return err;
 
-       err = ufs_mtk_unipro_set_pm(hba, 0);
+       err = ufs_mtk_unipro_set_pm(hba, false);
        if (err)
                return err;
 
 {
        int err;
 
-       err = ufs_mtk_unipro_set_pm(hba, 1);
+       err = ufs_mtk_unipro_set_pm(hba, true);
        if (err) {
                /* Resume UniPro state for following error recovery */
-               ufs_mtk_unipro_set_pm(hba, 0);
+               ufs_mtk_unipro_set_pm(hba, false);
                return err;
        }