1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/char/hw_random/octeon-rng.c b/drivers/char/hw_random/octeon-rng.c
index 1eada566ca70..f2885dbe1849 100644
--- a/drivers/char/hw_random/octeon-rng.c
+++ b/drivers/char/hw_random/octeon-rng.c
@@ -96,7 +96,7 @@ static int octeon_rng_probe(struct platform_device *pdev)
 
         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;
@@ -108,7 +108,7 @@ static int octeon_rng_probe(struct platform_device *pdev)
 
 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);