1 files changed, 6 insertions, 2 deletions

diff --git a/drivers/video/platinumfb.c b/drivers/video/platinumfb.c
index fdb33cd21a27..233871655824 100644
--- a/drivers/video/platinumfb.c
+++ b/drivers/video/platinumfb.c
@@ -34,6 +34,7 @@
 #include <asm/prom.h>
 #include <asm/pgtable.h>
 #include <asm/of_device.h>
+#include <asm/of_platform.h>
 
 #include "macmodes.h"
 #include "platinumfb.h"
@@ -626,6 +627,9 @@ static int __devinit platinumfb_probe(struct of_device* odev,
         
         rc = platinum_init_fb(info);
         if (rc != 0) {
+                iounmap(pinfo->frame_buffer);
+                iounmap(pinfo->platinum_regs);
+                iounmap(pinfo->cmap_regs);
                 dev_set_drvdata(&odev->dev, NULL);
                 framebuffer_release(info);
         }
@@ -682,14 +686,14 @@ static int __init platinumfb_init(void)
                 return -ENODEV;
         platinumfb_setup(option);
 #endif
-        of_register_driver(&platinum_driver);
+        of_register_platform_driver(&platinum_driver);
 
         return 0;
 }
 
 static void __exit platinumfb_exit(void)
 {
-        of_unregister_driver(&platinum_driver); 
+        of_unregister_platform_driver(&platinum_driver);
 }
 
 MODULE_LICENSE("GPL");