1 files changed, 2 insertions, 3 deletions

diff --git a/drivers/net/ethernet/sun/niu.c b/drivers/net/ethernet/sun/niu.c
index 95cff98d8a34..fa322409bff3 100644
--- a/drivers/net/ethernet/sun/niu.c
+++ b/drivers/net/ethernet/sun/niu.c
@@ -10108,7 +10108,7 @@ static int niu_of_probe(struct platform_device *op)
                 goto err_out_iounmap;
         }
 
-        dev_set_drvdata(&op->dev, dev);
+        platform_set_drvdata(op, dev);
 
         niu_device_announce(np);
 
@@ -10145,7 +10145,7 @@ err_out:
 
 static int niu_of_remove(struct platform_device *op)
 {
-        struct net_device *dev = dev_get_drvdata(&op->dev);
+        struct net_device *dev = platform_get_drvdata(op);
 
         if (dev) {
                 struct niu *np = netdev_priv(dev);
@@ -10175,7 +10175,6 @@ static int niu_of_remove(struct platform_device *op)
                 niu_put_parent(np);
 
                 free_netdev(dev);
-                dev_set_drvdata(&op->dev, NULL);
         }
         return 0;
 }