static int aspeed_video_probe(struct platform_device *pdev)
 {
+       struct aspeed_video *video;
        int rc;
-       struct resource *res;
-       struct aspeed_video *video =
-               devm_kzalloc(&pdev->dev, sizeof(*video), GFP_KERNEL);
 
+       video = devm_kzalloc(&pdev->dev, sizeof(*video), GFP_KERNEL);
        if (!video)
                return -ENOMEM;
 
+       video->base = devm_platform_ioremap_resource(pdev, 0);
+       if (IS_ERR(video->base))
+               return PTR_ERR(video->base);
+
        video->frame_rate = 30;
        video->dev = &pdev->dev;
        spin_lock_init(&video->lock);
        INIT_DELAYED_WORK(&video->res_work, aspeed_video_resolution_work);
        INIT_LIST_HEAD(&video->buffers);
 
-       res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
-
-       video->base = devm_ioremap_resource(video->dev, res);
-
-       if (IS_ERR(video->base))
-               return PTR_ERR(video->base);
-
        rc = aspeed_video_init(video);
        if (rc)
                return rc;