struct drm_fb_helper;
 
 #include <drm/drm_crtc.h>
+#include <drm/drm_device.h>
 #include <linux/kgdb.h>
 
 enum mode_set_atomic {
                       struct drm_fb_helper *helper,
                       int max_conn)
 {
+       /* So drivers can use it to free the struct */
+       helper->dev = dev;
+       dev->fb_helper = helper;
+
        return 0;
 }
 
 static inline void drm_fb_helper_fini(struct drm_fb_helper *helper)
 {
+       if (helper && helper->dev)
+               helper->dev->fb_helper = NULL;
 }
 
 static inline int drm_fb_helper_blank(int blank, struct fb_info *info)