};
 #endif
 
-static int __init omap2_mbox_probe(struct platform_device *pdev)
+static int __devinit omap2_mbox_probe(struct platform_device *pdev)
 {
        struct resource *res;
        int ret;
                goto err_dsp;
        }
 
-       ret = omap_mbox_register(&mbox_dsp_info);
+       ret = omap_mbox_register(&pdev->dev, &mbox_dsp_info);
        if (ret)
                goto err_dsp;
 
                        goto err_iva1;
                }
                mbox_iva_info.irq = res->start;
-               ret = omap_mbox_register(&mbox_iva_info);
+               ret = omap_mbox_register(&pdev->dev, &mbox_iva_info);
                if (ret)
                        goto err_iva1;
        }
        return ret;
 }
 
-static int omap2_mbox_remove(struct platform_device *pdev)
+static int __devexit omap2_mbox_remove(struct platform_device *pdev)
 {
 #if defined(CONFIG_ARCH_OMAP2420)
        omap_mbox_unregister(&mbox_iva_info);
 
 static struct platform_driver omap2_mbox_driver = {
        .probe = omap2_mbox_probe,
-       .remove = omap2_mbox_remove,
+       .remove = __devexit_p(omap2_mbox_remove),
        .driver = {
-               .name = "mailbox",
+               .name = "omap2-mailbox",
        },
 };
 
 MODULE_LICENSE("GPL v2");
 MODULE_DESCRIPTION("omap mailbox: omap2/3 architecture specific functions");
 MODULE_AUTHOR("Hiroshi DOYU <Hiroshi.DOYU@nokia.com>, Paul Mundt");
+MODULE_ALIAS("platform:omap2-mailbox");