#include <linux/slab.h>
 #include <linux/err.h>
 #include <linux/of.h>
+#include <linux/of_device.h>
 #include <linux/mfd/syscon.h>
 #include <linux/regmap.h>
 #include <linux/delay.h>
 static inline const struct s3c2410_wdt_variant *
 s3c2410_get_wdt_drv_data(struct platform_device *pdev)
 {
-       if (pdev->dev.of_node) {
-               const struct of_device_id *match;
-               match = of_match_node(s3c2410_wdt_match, pdev->dev.of_node);
-               return (struct s3c2410_wdt_variant *)match->data;
-       } else {
-               return (struct s3c2410_wdt_variant *)
-                       platform_get_device_id(pdev)->driver_data;
+       const struct s3c2410_wdt_variant *variant;
+
+       variant = of_device_get_match_data(&pdev->dev);
+       if (!variant) {
+               /* Device matched by platform_device_id */
+               variant = (struct s3c2410_wdt_variant *)
+                          platform_get_device_id(pdev)->driver_data;
        }
+
+       return variant;
 }
 
 static int s3c2410wdt_probe(struct platform_device *pdev)