break;
        }
        frmr->fr_nents = i;
+       frmr->fr_dir = direction;
 
        dma_nents = ib_dma_map_sg(device, frmr->fr_sg, frmr->fr_nents, direction);
        if (!dma_nents) {
        if (rc)
                goto out_senderr;
 
-       seg1->mr_dir = direction;
        seg1->rl_mw = mw;
        seg1->mr_rkey = mr->rkey;
        seg1->mr_base = mr->iova;
 
        seg->rl_mw = NULL;
 
-       ib_dma_unmap_sg(device, f->fr_sg, f->fr_nents, seg->mr_dir);
+       ib_dma_unmap_sg(device, f->fr_sg, f->fr_nents, f->fr_dir);
 
        if (!rc)
                rpcrdma_put_mw(r_xprt, mw);
        invalidate_wr->ex.invalidate_rkey = frmr->fr_mr->rkey;
        DECR_CQCOUNT(&r_xprt->rx_ep);
 
-       ib_dma_unmap_sg(ia->ri_device, frmr->fr_sg, frmr->fr_nents, seg1->mr_dir);
+       ib_dma_unmap_sg(ia->ri_device, frmr->fr_sg, frmr->fr_nents, frmr->fr_dir);
        read_lock(&ia->ri_qplock);
        rc = ib_post_send(ia->ri_id->qp, invalidate_wr, &bad_wr);
        read_unlock(&ia->ri_qplock);