if (err)
                goto out_free_units;
 
-       dev_set_drvdata(&op->dev, np);
+       platform_set_drvdata(op, np);
 
        schedule_delayed_work(&np->work, 0);
 
 
 static int n2rng_remove(struct platform_device *op)
 {
-       struct n2rng *np = dev_get_drvdata(&op->dev);
+       struct n2rng *np = platform_get_drvdata(op);
 
        np->flags |= N2RNG_FLAG_SHUTDOWN;
 
 
        kfree(np);
 
-       dev_set_drvdata(&op->dev, NULL);
-
        return 0;
 }
 
 
 
        rng->ops = ops;
 
-       dev_set_drvdata(&pdev->dev, &rng->ops);
+       platform_set_drvdata(pdev, &rng->ops);
        ret = hwrng_register(&rng->ops);
        if (ret)
                return -ENOENT;
 
 static int __exit octeon_rng_remove(struct platform_device *pdev)
 {
-       struct hwrng *rng = dev_get_drvdata(&pdev->dev);
+       struct hwrng *rng = platform_get_drvdata(pdev);
 
        hwrng_unregister(rng);
 
 
        };
 
        omap_rng_ops.priv = (unsigned long)priv;
-       dev_set_drvdata(&pdev->dev, priv);
+       platform_set_drvdata(pdev, priv);
 
        priv->mem_res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
        if (!priv->mem_res) {
                ret = PTR_ERR(priv->base);
                goto err_ioremap;
        }
-       dev_set_drvdata(&pdev->dev, priv);
+       platform_set_drvdata(pdev, priv);
 
        pm_runtime_enable(&pdev->dev);
        pm_runtime_get_sync(&pdev->dev);
 
 static int __exit omap_rng_remove(struct platform_device *pdev)
 {
-       struct omap_rng_private_data *priv = dev_get_drvdata(&pdev->dev);
+       struct omap_rng_private_data *priv = platform_get_drvdata(pdev);
 
        hwrng_unregister(&omap_rng_ops);