static void __iomem *sysc_base2;
 
-static int rmobile_reset_handler(struct notifier_block *this,
-                                unsigned long mode, void *cmd)
+static int rmobile_reset_handler(struct sys_off_data *data)
 {
-       pr_debug("%s %lu\n", __func__, mode);
-
        /* Let's assume we have acquired the HPB semaphore */
        writel(RESCNT2_PRES, sysc_base2 + RESCNT2);
 
        return NOTIFY_DONE;
 }
 
-static struct notifier_block rmobile_reset_nb = {
-       .notifier_call = rmobile_reset_handler,
-       .priority = 192,
-};
-
 static int rmobile_reset_probe(struct platform_device *pdev)
 {
        int error;
        if (IS_ERR(sysc_base2))
                return PTR_ERR(sysc_base2);
 
-       error = register_restart_handler(&rmobile_reset_nb);
+       error = devm_register_sys_off_handler(&pdev->dev,
+                                             SYS_OFF_MODE_RESTART,
+                                             SYS_OFF_PRIO_HIGH,
+                                             rmobile_reset_handler,
+                                             NULL);
        if (error) {
                dev_err(&pdev->dev,
                        "cannot register restart handler (err=%d)\n", error);
        return 0;
 }
 
-static void rmobile_reset_remove(struct platform_device *pdev)
-{
-       unregister_restart_handler(&rmobile_reset_nb);
-}
-
 static const struct of_device_id rmobile_reset_of_match[] = {
        { .compatible = "renesas,sysc-rmobile", },
        { /* sentinel */ }
 
 static struct platform_driver rmobile_reset_driver = {
        .probe = rmobile_reset_probe,
-       .remove_new = rmobile_reset_remove,
        .driver = {
                .name = "rmobile_reset",
                .of_match_table = rmobile_reset_of_match,