1 files changed, 2 insertions, 4 deletions

diff --git a/drivers/net/ethernet/ibm/emac/mal.c b/drivers/net/ethernet/ibm/emac/mal.c
index 610ed223d1db..856ea66c9223 100644
--- a/drivers/net/ethernet/ibm/emac/mal.c
+++ b/drivers/net/ethernet/ibm/emac/mal.c
@@ -696,7 +696,7 @@ static int mal_probe(struct platform_device *ofdev)
 
         /* Advertise this instance to the rest of the world */
         wmb();
-        dev_set_drvdata(&ofdev->dev, mal);
+        platform_set_drvdata(ofdev, mal);
 
         mal_dbg_register(mal);
 
@@ -722,7 +722,7 @@ static int mal_probe(struct platform_device *ofdev)
 
 static int mal_remove(struct platform_device *ofdev)
 {
-        struct mal_instance *mal = dev_get_drvdata(&ofdev->dev);
+        struct mal_instance *mal = platform_get_drvdata(ofdev);
 
         MAL_DBG(mal, "remove" NL);
 
@@ -735,8 +735,6 @@ static int mal_remove(struct platform_device *ofdev)
                        "mal%d: commac list is not empty on remove!\n",
                        mal->index);
 
-        dev_set_drvdata(&ofdev->dev, NULL);
-
         free_irq(mal->serr_irq, mal);
         free_irq(mal->txde_irq, mal);
         free_irq(mal->txeob_irq, mal);