#include <linux/module.h>
 #include <linux/notifier.h>
+#include <linux/of_platform.h>
 #include <linux/platform_device.h>
 
 #include <drm/drmP.h>
 #define AWG_DELAY_ED        (-8)
 #define AWG_DELAY_SD        (-7)
 
-static LIST_HEAD(vtg_lookup);
-
 /*
  * STI VTG register offset structure
  *
 /**
  * STI VTG structure
  *
- * @dev: pointer to device driver
- * @np: device node
  * @regs: register mapping
  * @sync_params: synchronisation parameters used to generate timings
  * @irq: VTG irq
  * @irq_status: store the IRQ status value
  * @notifier_list: notifier callback
  * @crtc: the CRTC for vblank event
- * @link: List node to link the structure in lookup list
  */
 struct sti_vtg {
-       struct device *dev;
-       struct device_node *np;
        void __iomem *regs;
        struct sti_vtg_sync_params sync_params[VTG_MAX_SYNC_OUTPUT];
        int irq;
        u32 irq_status;
        struct raw_notifier_head notifier_list;
        struct drm_crtc *crtc;
-       struct list_head link;
 };
 
-static void vtg_register(struct sti_vtg *vtg)
-{
-       list_add_tail(&vtg->link, &vtg_lookup);
-}
-
 struct sti_vtg *of_vtg_find(struct device_node *np)
 {
-       struct sti_vtg *vtg;
+       struct platform_device *pdev;
 
-       list_for_each_entry(vtg, &vtg_lookup, link) {
-               if (vtg->np == np)
-                       return vtg;
-       }
-       return NULL;
+       pdev = of_find_device_by_node(np);
+       if (!pdev)
+               return NULL;
+
+       return (struct sti_vtg *)platform_get_drvdata(pdev);
 }
 
 static void vtg_reset(struct sti_vtg *vtg)
        if (!vtg)
                return -ENOMEM;
 
-       vtg->dev = dev;
-       vtg->np = pdev->dev.of_node;
-
        /* Get Memory ressources */
        res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
        if (!res) {
                return ret;
        }
 
-       vtg_register(vtg);
        platform_set_drvdata(pdev, vtg);
 
-       DRM_INFO("%s %s\n", __func__, dev_name(vtg->dev));
+       DRM_INFO("%s %s\n", __func__, dev_name(dev));
 
        return 0;
 }