1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/sbus/char/flash.c b/drivers/sbus/char/flash.c
index 368d66294d83..d79f386c3486 100644
--- a/drivers/sbus/char/flash.c
+++ b/drivers/sbus/char/flash.c
@@ -218,12 +218,12 @@ static struct of_platform_driver flash_driver = {
 
 static int __init flash_init(void)
 {
-        return of_register_driver(&flash_driver, &of_bus_type);
+        return of_register_platform_driver(&flash_driver);
 }
 
 static void __exit flash_cleanup(void)
 {
-        of_unregister_driver(&flash_driver);
+        of_unregister_platform_driver(&flash_driver);
 }
 
 module_init(flash_init);