__raw_writel(ctrl & ~RNGA_CONTROL_GO, mxc_rng->mem + RNGA_CONTROL);
 }
 
-static int __init mxc_rnga_probe(struct platform_device *pdev)
+static int mxc_rnga_probe(struct platform_device *pdev)
 {
        int err;
        struct mxc_rng *mxc_rng;
        return err;
 }
 
-static void __exit mxc_rnga_remove(struct platform_device *pdev)
+static void mxc_rnga_remove(struct platform_device *pdev)
 {
        struct mxc_rng *mxc_rng = platform_get_drvdata(pdev);
 
                .name = "mxc_rnga",
                .of_match_table = mxc_rnga_of_match,
        },
-       .remove_new = __exit_p(mxc_rnga_remove),
+       .probe = mxc_rnga_probe,
+       .remove_new = mxc_rnga_remove,
 };
 
-module_platform_driver_probe(mxc_rnga_driver, mxc_rnga_probe);
+module_platform_driver(mxc_rnga_driver);
 
 MODULE_AUTHOR("Freescale Semiconductor, Inc.");
 MODULE_DESCRIPTION("H/W RNGA driver for i.MX");