int ret = 0;
 
        if (drm_gem_is_imported(obj)) {
-               ret = dma_buf_vmap(obj->import_attach->dmabuf, map);
+               ret = dma_buf_vmap(obj->dma_buf, map);
                if (!ret) {
                        if (drm_WARN_ON(obj->dev, map->is_iomem)) {
-                               dma_buf_vunmap(obj->import_attach->dmabuf, map);
+                               dma_buf_vunmap(obj->dma_buf, map);
                                return -EIO;
                        }
                }
        struct drm_gem_object *obj = &shmem->base;
 
        if (drm_gem_is_imported(obj)) {
-               dma_buf_vunmap(obj->import_attach->dmabuf, map);
+               dma_buf_vunmap(obj->dma_buf, map);
        } else {
                dma_resv_assert_held(shmem->base.resv);