.resume         = ledtrig_cpu_syscore_resume,
 };
 
-static int ledtrig_starting_cpu(unsigned int cpu)
+static int ledtrig_online_cpu(unsigned int cpu)
 {
        ledtrig_cpu(CPU_LED_START);
        return 0;
 }
 
-static int ledtrig_dying_cpu(unsigned int cpu)
+static int ledtrig_prepare_down_cpu(unsigned int cpu)
 {
        ledtrig_cpu(CPU_LED_STOP);
        return 0;
 static int __init ledtrig_cpu_init(void)
 {
        int cpu;
+       int ret;
 
        /* Supports up to 9999 cpu cores */
        BUILD_BUG_ON(CONFIG_NR_CPUS > 9999);
 
        register_syscore_ops(&ledtrig_cpu_syscore_ops);
 
-       /*
-        * FIXME: Why needs this to happen in the interrupt disabled
-        * low level bringup phase of a cpu?
-        */
-       cpuhp_setup_state(CPUHP_AP_LEDTRIG_STARTING, "AP_LEDTRIG_STARTING",
-                         ledtrig_starting_cpu, ledtrig_dying_cpu);
+       ret = cpuhp_setup_state(CPUHP_AP_ONLINE_DYN, "AP_LEDTRIG_STARTING",
+                               ledtrig_online_cpu, ledtrig_prepare_down_cpu);
+       if (ret < 0)
+               pr_err("CPU hotplug notifier for ledtrig-cpu could not be registered: %d\n",
+                      ret);
 
        pr_info("ledtrig-cpu: registered to indicate activity on CPUs\n");
 
 
        CPUHP_AP_ARM_CORESIGHT_STARTING,
        CPUHP_AP_ARM_CORESIGHT4_STARTING,
        CPUHP_AP_ARM64_ISNDEP_STARTING,
-       CPUHP_AP_LEDTRIG_STARTING,
        CPUHP_AP_SMPCFD_DYING,
        CPUHP_AP_X86_TBOOT_DYING,
        CPUHP_AP_NOTIFY_STARTING,