*/
 #include <linux/clk.h>
 #include <linux/io.h>
-#include <linux/init.h>
+#include <linux/module.h>
 #include <linux/of_irq.h>
 #include <linux/of_platform.h>
 #include <linux/platform_device.h>
        return 0;
 }
 
+static int mtk8250_remove(struct platform_device *pdev)
+{
+       struct mtk8250_data *data = platform_get_drvdata(pdev);
+
+       pm_runtime_get_sync(&pdev->dev);
+
+       serial8250_unregister_port(data->line);
+
+       pm_runtime_disable(&pdev->dev);
+       pm_runtime_put_noidle(&pdev->dev);
+
+       if (!pm_runtime_status_suspended(&pdev->dev))
+               mtk8250_runtime_suspend(&pdev->dev);
+
+       return 0;
+}
+
 static int __maybe_unused mtk8250_suspend(struct device *dev)
 {
        struct mtk8250_data *data = dev_get_drvdata(dev);
        { .compatible = "mediatek,mt6577-uart" },
        { /* Sentinel */ }
 };
+MODULE_DEVICE_TABLE(of, mtk8250_of_match);
 
 static struct platform_driver mtk8250_platform_driver = {
        .driver = {
-               .name                   = "mt6577-uart",
-               .pm                     = &mtk8250_pm_ops,
-               .of_match_table         = mtk8250_of_match,
-               .suppress_bind_attrs    = true,
-
+               .name           = "mt6577-uart",
+               .pm             = &mtk8250_pm_ops,
+               .of_match_table = mtk8250_of_match,
        },
        .probe                  = mtk8250_probe,
+       .remove                 = mtk8250_remove,
 };
-builtin_platform_driver(mtk8250_platform_driver);
+module_platform_driver(mtk8250_platform_driver);
 
 #ifdef CONFIG_SERIAL_8250_CONSOLE
 static int __init early_mtk8250_setup(struct earlycon_device *device,
 
 OF_EARLYCON_DECLARE(mtk8250, "mediatek,mt6577-uart", early_mtk8250_setup);
 #endif
+
+MODULE_AUTHOR("Matthias Brugger");
+MODULE_LICENSE("GPL");
+MODULE_DESCRIPTION("Mediatek 8250 serial port driver");