static void ufs_qcom_enable_test_bus(struct ufs_qcom_host *host)
 {
-       if (host->dbg_print_en & UFS_QCOM_DBG_PRINT_TEST_BUS_EN)
+       if (host->dbg_print_en & UFS_QCOM_DBG_PRINT_TEST_BUS_EN) {
+               ufshcd_rmwl(host->hba, UFS_REG_TEST_BUS_EN,
+                               UFS_REG_TEST_BUS_EN, REG_UFS_CFG1);
                ufshcd_rmwl(host->hba, TEST_BUS_EN, TEST_BUS_EN, REG_UFS_CFG1);
-       else
+       } else {
+               ufshcd_rmwl(host->hba, UFS_REG_TEST_BUS_EN, 0, REG_UFS_CFG1);
                ufshcd_rmwl(host->hba, TEST_BUS_EN, 0, REG_UFS_CFG1);
+       }
 }
 
 static void ufs_qcom_get_default_testbus_cfg(struct ufs_qcom_host *host)
 {
        /* provide a legal default configuration */
-       host->testbus.select_major = TSTBUS_UAWM;
-       host->testbus.select_minor = 1;
+       host->testbus.select_major = TSTBUS_UNIPRO;
+       host->testbus.select_minor = 37;
 }
 
 static bool ufs_qcom_testbus_cfg_is_ok(struct ufs_qcom_host *host)
         * mappings of select_minor, since there is no harm in
         * configuring a non-existent select_minor
         */
-       if (host->testbus.select_minor > 0x1F) {
+       if (host->testbus.select_minor > 0xFF) {
                dev_err(host->hba->dev,
                        "%s: 0x%05X is not a legal testbus option\n",
                        __func__, host->testbus.select_minor);
                break;
        case TSTBUS_UNIPRO:
                reg = UFS_UNIPRO_CFG;
-               offset = 1;
+               offset = 20;
+               mask = 0xFFF;
                break;
        /*
         * No need for a default case, since
                    (u32)host->testbus.select_minor << offset,
                    reg);
        ufs_qcom_enable_test_bus(host);
+       /*
+        * Make sure the test bus configuration is
+        * committed before returning.
+        */
+       mb();
        ufshcd_release(host->hba);
        pm_runtime_put_sync(host->hba->dev);
 
        ufs_qcom_dump_regs(hba, UFS_TEST_BUS, 1, "UFS_TEST_BUS ");
 }
 
+static void ufs_qcom_print_unipro_testbus(struct ufs_hba *hba)
+{
+       struct ufs_qcom_host *host = ufshcd_get_variant(hba);
+       u32 *testbus = NULL;
+       int i, nminor = 256, testbus_len = nminor * sizeof(u32);
+
+       testbus = kmalloc(testbus_len, GFP_KERNEL);
+       if (!testbus)
+               return;
+
+       host->testbus.select_major = TSTBUS_UNIPRO;
+       for (i = 0; i < nminor; i++) {
+               host->testbus.select_minor = i;
+               ufs_qcom_testbus_config(host);
+               testbus[i] = ufshcd_readl(hba, UFS_TEST_BUS);
+       }
+       print_hex_dump(KERN_ERR, "UNIPRO_TEST_BUS ", DUMP_PREFIX_OFFSET,
+                       16, 4, testbus, testbus_len, false);
+       kfree(testbus);
+}
+
 static void ufs_qcom_dump_dbg_regs(struct ufs_hba *hba)
 {
        ufs_qcom_dump_regs(hba, REG_UFS_SYS1CLK_1US, 16,
                        "HCI Vendor Specific Registers ");
 
+       /* sleep a bit intermittently as we are dumping too much data */
        ufs_qcom_print_hw_debug_reg_all(hba, NULL, ufs_qcom_dump_regs_wrapper);
+       usleep_range(1000, 1100);
        ufs_qcom_testbus_read(hba);
+       usleep_range(1000, 1100);
+       ufs_qcom_print_unipro_testbus(hba);
+       usleep_range(1000, 1100);
 }
 
 /**