#include <drm/drm_gem_atomic_helper.h>
 #include <drm/drm_gem_framebuffer_helper.h>
 #include <drm/drm_plane_helper.h>
-#include <drm/drm_gem_shmem_helper.h>
 
 #include "vkms_drv.h"
 
        return 0;
 }
 
-static int vkms_prepare_fb(struct drm_plane *plane,
-                          struct drm_plane_state *state)
-{
-       struct drm_gem_object *gem_obj;
-       struct dma_buf_map map;
-       int ret;
-
-       if (!state->fb)
-               return 0;
-
-       gem_obj = drm_gem_fb_get_obj(state->fb, 0);
-       ret = drm_gem_shmem_vmap(gem_obj, &map);
-       if (ret)
-               DRM_ERROR("vmap failed: %d\n", ret);
-
-       return drm_gem_plane_helper_prepare_fb(plane, state);
-}
-
-static void vkms_cleanup_fb(struct drm_plane *plane,
-                           struct drm_plane_state *old_state)
-{
-       struct drm_gem_object *gem_obj;
-       struct drm_gem_shmem_object *shmem_obj;
-       struct dma_buf_map map;
-
-       if (!old_state->fb)
-               return;
-
-       gem_obj = drm_gem_fb_get_obj(old_state->fb, 0);
-       shmem_obj = to_drm_gem_shmem_obj(drm_gem_fb_get_obj(old_state->fb, 0));
-       dma_buf_map_set_vaddr(&map, shmem_obj->vaddr);
-       drm_gem_shmem_vunmap(gem_obj, &map);
-}
-
 static const struct drm_plane_helper_funcs vkms_primary_helper_funcs = {
        .atomic_update          = vkms_plane_atomic_update,
        .atomic_check           = vkms_plane_atomic_check,
-       .prepare_fb             = vkms_prepare_fb,
-       .cleanup_fb             = vkms_cleanup_fb,
+       DRM_GEM_SHADOW_PLANE_HELPER_FUNCS,
 };
 
 struct vkms_plane *vkms_plane_init(struct vkms_device *vkmsdev,