if (device_may_wakeup(dev)) {
                enable_irq_wake(kp->key_sense_irq);
        } else {
-               mutex_lock(&input_dev->mutex);
+               guard(mutex)(&input_dev->mutex);
 
                if (input_device_enabled(input_dev))
                        pmic8xxx_kp_disable(kp);
-
-               mutex_unlock(&input_dev->mutex);
        }
 
        return 0;
        if (device_may_wakeup(dev)) {
                disable_irq_wake(kp->key_sense_irq);
        } else {
-               mutex_lock(&input_dev->mutex);
+               guard(mutex)(&input_dev->mutex);
 
                if (input_device_enabled(input_dev))
                        pmic8xxx_kp_enable(kp);
-
-               mutex_unlock(&input_dev->mutex);
        }
 
        return 0;