* To avoid these unnecessary/illegal step we skip to the last error
         * handling stage: reset and restore.
         */
-       if (lrbp->lun == UFS_UPIU_UFS_DEVICE_WLUN)
+       if (lrbp->lun == UFS_UPIU_UFS_DEVICE_WLUN) {
+               ufshcd_update_reg_hist(&hba->ufs_stats.task_abort, lrbp->lun);
                return ufshcd_eh_host_reset_handler(cmd);
+       }
 
        ufshcd_hold(hba, false);
        reg = ufshcd_readl(hba, REG_UTP_TRANSFER_REQ_DOOR_BELL);
         */
        scsi_print_command(hba->lrb[tag].cmd);
        if (!hba->req_abort_count) {
-               ufshcd_update_reg_hist(&hba->ufs_stats.task_abort, 0);
+               ufshcd_update_reg_hist(&hba->ufs_stats.task_abort, tag);
                ufshcd_print_host_regs(hba);
                ufshcd_print_host_state(hba);
                ufshcd_print_pwr_info(hba);