1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/watchdog/riowd.c b/drivers/watchdog/riowd.c
index 5dceeddc8859..4082b4ace1fc 100644
--- a/drivers/watchdog/riowd.c
+++ b/drivers/watchdog/riowd.c
@@ -250,12 +250,12 @@ static struct of_platform_driver riowd_driver = {
 
 static int __init riowd_init(void)
 {
-        return of_register_driver(&riowd_driver, &of_bus_type);
+        return of_register_platform_driver(&riowd_driver);
 }
 
 static void __exit riowd_exit(void)
 {
-        of_unregister_driver(&riowd_driver);
+        of_unregister_platform_driver(&riowd_driver);
 }
 
 module_init(riowd_init);