#define AMBA_CONSOLE   NULL
 #endif
 
+static DEFINE_MUTEX(amba_reg_lock);
 static struct uart_driver amba_reg = {
        .owner                  = THIS_MODULE,
        .driver_name            = "ttyAM",
        amba_ports[i] = uap;
 
        amba_set_drvdata(dev, uap);
+
+       mutex_lock(&amba_reg_lock);
+       if (!amba_reg.state) {
+               ret = uart_register_driver(&amba_reg);
+               if (ret < 0) {
+                       mutex_unlock(&amba_reg_lock);
+                       dev_err(uap->port.dev,
+                               "Failed to register AMBA-PL010 driver\n");
+                       return ret;
+               }
+       }
+       mutex_unlock(&amba_reg_lock);
+
        ret = uart_add_one_port(&amba_reg, &uap->port);
        if (ret)
                amba_ports[i] = NULL;
 {
        struct uart_amba_port *uap = amba_get_drvdata(dev);
        int i;
+       bool busy = false;
 
        uart_remove_one_port(&amba_reg, &uap->port);
 
        for (i = 0; i < ARRAY_SIZE(amba_ports); i++)
                if (amba_ports[i] == uap)
                        amba_ports[i] = NULL;
+               else if (amba_ports[i])
+                       busy = true;
+
+       if (!busy)
+               uart_unregister_driver(&amba_reg);
 
        return 0;
 }
 
 static int __init pl010_init(void)
 {
-       int ret;
-
        printk(KERN_INFO "Serial: AMBA driver\n");
 
-       ret = uart_register_driver(&amba_reg);
-       if (ret == 0) {
-               ret = amba_driver_register(&pl010_driver);
-               if (ret)
-                       uart_unregister_driver(&amba_reg);
-       }
-       return ret;
+       return  amba_driver_register(&pl010_driver);
 }
 
 static void __exit pl010_exit(void)
 {
        amba_driver_unregister(&pl010_driver);
-       uart_unregister_driver(&amba_reg);
 }
 
 module_init(pl010_init);