1 files changed, 6 insertions, 3 deletions

diff --git a/drivers/i2c/busses/i2c-s3c2410.c b/drivers/i2c/busses/i2c-s3c2410.c
index 6ced28e90070..1b582262e677 100644
--- a/drivers/i2c/busses/i2c-s3c2410.c
+++ b/drivers/i2c/busses/i2c-s3c2410.c
@@ -33,7 +33,7 @@
 #include <linux/delay.h>
 #include <linux/errno.h>
 #include <linux/err.h>
-#include <linux/device.h>
+#include <linux/platform_device.h>
 
 #include <asm/hardware.h>
 #include <asm/irq.h>
@@ -918,8 +918,11 @@ static int __init i2c_adap_s3c_init(void)
         int ret;
 
         ret = driver_register(&s3c2410_i2c_driver);
-        if (ret == 0)
-                ret = driver_register(&s3c2440_i2c_driver); 
+        if (ret == 0) {
+                ret = driver_register(&s3c2440_i2c_driver);
+                if (ret)
+                        driver_unregister(&s3c2410_i2c_driver);
+        }
 
         return ret;
 }