1 files changed, 12 insertions, 3 deletions

diff --git a/drivers/char/ttyprintk.c b/drivers/char/ttyprintk.c
index daea84c41743..a15ce4ef39cd 100644
--- a/drivers/char/ttyprintk.c
+++ b/drivers/char/ttyprintk.c
@@ -17,7 +17,7 @@
 #include <linux/device.h>
 #include <linux/serial.h>
 #include <linux/tty.h>
-#include <linux/export.h>
+#include <linux/module.h>
 
 struct ttyprintk_port {
         struct tty_port port;
@@ -210,10 +210,19 @@ static int __init ttyprintk_init(void)
         return 0;
 
 error:
-        tty_unregister_driver(ttyprintk_driver);
         put_tty_driver(ttyprintk_driver);
         tty_port_destroy(&tpk_port.port);
-        ttyprintk_driver = NULL;
         return ret;
 }
+
+static void __exit ttyprintk_exit(void)
+{
+        tty_unregister_driver(ttyprintk_driver);
+        put_tty_driver(ttyprintk_driver);
+        tty_port_destroy(&tpk_port.port);
+}
+
 device_initcall(ttyprintk_init);
+module_exit(ttyprintk_exit);
+
+MODULE_LICENSE("GPL");