return;
 
        (void)ntb_mw_clear_trans(peer->perf->ntb, peer->pidx, peer->gidx);
-       dma_free_coherent(&peer->perf->ntb->dev, peer->inbuf_size,
+       dma_free_coherent(&peer->perf->ntb->pdev->dev, peer->inbuf_size,
                          peer->inbuf, peer->inbuf_xlat);
        peer->inbuf = NULL;
 }
 
        perf_free_inbuf(peer);
 
-       peer->inbuf = dma_alloc_coherent(&perf->ntb->dev, peer->inbuf_size,
-                                        &peer->inbuf_xlat, GFP_KERNEL);
+       peer->inbuf = dma_alloc_coherent(&perf->ntb->pdev->dev,
+                                        peer->inbuf_size, &peer->inbuf_xlat,
+                                        GFP_KERNEL);
        if (!peer->inbuf) {
                dev_err(&perf->ntb->dev, "Failed to alloc inbuf of %pa\n",
                        &peer->inbuf_size);
        destroy_workqueue(perf_wq);
 }
 module_exit(perf_exit);
-