#include <linux/mtd/mtd.h>
 #include <linux/mux/consumer.h>
 #include <linux/of.h>
+#include <linux/of_address.h>
 #include <linux/platform_device.h>
 #include <linux/pm_runtime.h>
 #include <linux/types.h>
 
 static int am654_hbmc_probe(struct platform_device *pdev)
 {
+       struct device_node *np = pdev->dev.of_node;
        struct device *dev = &pdev->dev;
        struct am654_hbmc_priv *priv;
+       struct resource res;
        int ret;
 
        priv = devm_kzalloc(dev, sizeof(*priv), GFP_KERNEL);
 
        platform_set_drvdata(pdev, priv);
 
+       ret = of_address_to_resource(np, 0, &res);
+       if (ret)
+               return ret;
+
        if (of_property_read_bool(dev->of_node, "mux-controls")) {
                struct mux_control *control = devm_mux_control_get(dev, NULL);
 
                goto disable_pm;
        }
 
+       priv->hbdev.map.size = resource_size(&res);
+       priv->hbdev.map.virt = devm_ioremap_resource(dev, &res);
+       if (IS_ERR(priv->hbdev.map.virt))
+               return PTR_ERR(priv->hbdev.map.virt);
+
        priv->ctlr.dev = dev;
        priv->ctlr.ops = &am654_hbmc_ops;
        priv->hbdev.ctlr = &priv->ctlr;
 
 #include <linux/mtd/map.h>
 #include <linux/mtd/mtd.h>
 #include <linux/of.h>
-#include <linux/of_address.h>
 #include <linux/types.h>
 
 static struct hyperbus_device *map_to_hbdev(struct map_info *map)
        struct hyperbus_ctlr *ctlr;
        struct device_node *np;
        struct map_info *map;
-       struct resource res;
        struct device *dev;
        int ret;
 
 
        hbdev->memtype = HYPERFLASH;
 
-       ret = of_address_to_resource(np, 0, &res);
-       if (ret)
-               return ret;
-
        dev = ctlr->dev;
        map = &hbdev->map;
-       map->size = resource_size(&res);
-       map->virt = devm_ioremap_resource(dev, &res);
-       if (IS_ERR(map->virt))
-               return PTR_ERR(map->virt);
-
        map->name = dev_name(dev);
        map->bankwidth = 2;
        map->device_node = np;