{
        struct acpi_device *acpi_device = ACPI_COMPANION(&device->dev);
 
-       acpi_remove_notify_handler(acpi_device->handle, ACPI_DEVICE_NOTIFY,
+       acpi_remove_notify_handler(acpi_device->handle, ACPI_ALL_NOTIFY,
                                   acpi_wmi_notify_handler);
        acpi_remove_address_space_handler(acpi_device->handle,
                                ACPI_ADR_SPACE_EC, &acpi_wmi_ec_space_handler);
        }
 
        status = acpi_install_notify_handler(acpi_device->handle,
-                                            ACPI_DEVICE_NOTIFY,
+                                            ACPI_ALL_NOTIFY,
                                             acpi_wmi_notify_handler,
                                             NULL);
        if (ACPI_FAILURE(status)) {
        device_unregister(wmi_bus_dev);
 
 err_remove_notify_handler:
-       acpi_remove_notify_handler(acpi_device->handle, ACPI_DEVICE_NOTIFY,
+       acpi_remove_notify_handler(acpi_device->handle, ACPI_ALL_NOTIFY,
                                   acpi_wmi_notify_handler);
 
 err_remove_ec_handler: