tc->reserved_E8_0 = 0;
 
        if ((type & SCSI_PROT_DIF_TYPE1) || (type & SCSI_PROT_DIF_TYPE2))
-               tc->ref_tag_seed_gen = scsi_get_lba(scmd) & 0xffffffff;
+               tc->ref_tag_seed_gen = scsi_prot_ref_tag(scmd);
        else if (type & SCSI_PROT_DIF_TYPE3)
                tc->ref_tag_seed_gen = 0;
 }
        tc->app_tag_gen = 0;
 
        if ((type & SCSI_PROT_DIF_TYPE1) || (type & SCSI_PROT_DIF_TYPE2))
-               tc->ref_tag_seed_verify = scsi_get_lba(scmd) & 0xffffffff;
+               tc->ref_tag_seed_verify = scsi_prot_ref_tag(scmd);
        else if (type & SCSI_PROT_DIF_TYPE3)
                tc->ref_tag_seed_verify = 0;