/* Take a command buffer for each screen from the end of usable VRAM. */
        vbox->available_vram_size -= vbox->num_crtcs * VBVA_MIN_BUFFER_SIZE;
 
-       vbox->vbva_buffers = pci_iomap_range(pdev, 0,
-                                            vbox->available_vram_size,
-                                            vbox->num_crtcs *
-                                            VBVA_MIN_BUFFER_SIZE);
-       if (!vbox->vbva_buffers)
-               return -ENOMEM;
+       vbox->vbva_buffers = pcim_iomap_range(
+                       pdev, 0, vbox->available_vram_size,
+                       vbox->num_crtcs * VBVA_MIN_BUFFER_SIZE);
+       if (IS_ERR(vbox->vbva_buffers))
+               return PTR_ERR(vbox->vbva_buffers);
 
        for (i = 0; i < vbox->num_crtcs; ++i) {
                vbva_setup_buffer_context(&vbox->vbva_info[i],
        DRM_INFO("VRAM %08x\n", vbox->full_vram_size);
 
        /* Map guest-heap at end of vram */
-       vbox->guest_heap =
-           pci_iomap_range(pdev, 0, GUEST_HEAP_OFFSET(vbox),
-                           GUEST_HEAP_SIZE);
-       if (!vbox->guest_heap)
-               return -ENOMEM;
+       vbox->guest_heap = pcim_iomap_range(pdev, 0,
+                       GUEST_HEAP_OFFSET(vbox), GUEST_HEAP_SIZE);
+       if (IS_ERR(vbox->guest_heap))
+               return PTR_ERR(vbox->guest_heap);
 
        /* Create guest-heap mem-pool use 2^4 = 16 byte chunks */
        vbox->guest_pool = devm_gen_pool_create(vbox->ddev.dev, 4, -1,