#include <linux/pm.h>
 
 static void __iomem *msm_ps_hold;
-static int deassert_pshold(struct notifier_block *nb, unsigned long action,
-                          void *data)
-{
-       writel(0, msm_ps_hold);
-       mdelay(10000);
-
-       return NOTIFY_DONE;
-}
-
-static struct notifier_block restart_nb = {
-       .notifier_call = deassert_pshold,
-       .priority = 128,
-};
 
 static int do_msm_poweroff(struct sys_off_data *data)
 {
-       deassert_pshold(&restart_nb, 0, NULL);
+       writel(0, msm_ps_hold);
+       mdelay(10000);
 
        return NOTIFY_DONE;
 }
        if (IS_ERR(msm_ps_hold))
                return PTR_ERR(msm_ps_hold);
 
-       register_restart_handler(&restart_nb);
+       devm_register_sys_off_handler(&pdev->dev, SYS_OFF_MODE_RESTART,
+                                     128, do_msm_poweroff, NULL);
 
        devm_register_sys_off_handler(&pdev->dev, SYS_OFF_MODE_POWER_OFF,
                                      SYS_OFF_PRIO_DEFAULT, do_msm_poweroff,