#include <linux/errno.h>
 #include <linux/fs.h>
 #include <linux/init.h>
+#include <linux/io.h>
 #include <linux/kernel.h>
 #include <linux/miscdevice.h>
 #include <linux/module.h>
                return -ENODEV;
        }
 
-       bcm63xx_wdt_device.regs = ioremap_nocache(r->start, resource_size(r));
+       bcm63xx_wdt_device.regs = devm_ioremap_nocache(&pdev->dev, r->start,
+                                                       resource_size(r));
        if (!bcm63xx_wdt_device.regs) {
                dev_err(&pdev->dev, "failed to remap I/O resources\n");
                return -ENXIO;
        ret = bcm63xx_timer_register(TIMER_WDT_ID, bcm63xx_wdt_isr, NULL);
        if (ret < 0) {
                dev_err(&pdev->dev, "failed to register wdt timer isr\n");
-               goto unmap;
+               return ret;
        }
 
        if (bcm63xx_wdt_settimeout(wdt_time)) {
 
 unregister_timer:
        bcm63xx_timer_unregister(TIMER_WDT_ID);
-unmap:
-       iounmap(bcm63xx_wdt_device.regs);
        return ret;
 }
 
 
        misc_deregister(&bcm63xx_wdt_miscdev);
        bcm63xx_timer_unregister(TIMER_WDT_ID);
-       iounmap(bcm63xx_wdt_device.regs);
        return 0;
 }