1 files changed, 2 insertions, 4 deletions

diff --git a/drivers/tty/serial/sunzilog.c b/drivers/tty/serial/sunzilog.c
index 813ef8eb8eff..135a15203532 100644
--- a/drivers/tty/serial/sunzilog.c
+++ b/drivers/tty/serial/sunzilog.c
@@ -1495,7 +1495,7 @@ static int zs_probe(struct platform_device *op)
                 kbm_inst++;
         }
 
-        dev_set_drvdata(&op->dev, &up[0]);
+        platform_set_drvdata(op, &up[0]);
 
         return 0;
 }
@@ -1512,7 +1512,7 @@ static void zs_remove_one(struct uart_sunzilog_port *up)
 
 static int zs_remove(struct platform_device *op)
 {
-        struct uart_sunzilog_port *up = dev_get_drvdata(&op->dev);
+        struct uart_sunzilog_port *up = platform_get_drvdata(op);
         struct zilog_layout __iomem *regs;
 
         zs_remove_one(&up[0]);
@@ -1521,8 +1521,6 @@ static int zs_remove(struct platform_device *op)
         regs = sunzilog_chip_regs[up[0].port.line / 2];
         of_iounmap(&op->resource[0], regs, sizeof(struct zilog_layout));
 
-        dev_set_drvdata(&op->dev, NULL);
-
         return 0;
 }