return error;
 
        /* Lock the device to prevent races with open/close (and itself) */
-       mutex_lock(&input_dev->mutex);
-
-       disable_irq(client->irq);
+       guard(mutex)(&input_dev->mutex);
+       guard(disable_irq)(&client->irq);
 
        /*
         * Set current interval to the greater of the minimum interval or
 
        kxtj9_update_odr(tj9, tj9->last_poll_interval);
 
-       enable_irq(client->irq);
-       mutex_unlock(&input_dev->mutex);
-
        return count;
 }
 
        struct kxtj9_data *tj9 = i2c_get_clientdata(client);
        struct input_dev *input_dev = tj9->input_dev;
 
-       mutex_lock(&input_dev->mutex);
+       guard(mutex)(&input_dev->mutex);
 
        if (input_device_enabled(input_dev))
                kxtj9_disable(tj9);
 
-       mutex_unlock(&input_dev->mutex);
        return 0;
 }
 
        struct kxtj9_data *tj9 = i2c_get_clientdata(client);
        struct input_dev *input_dev = tj9->input_dev;
 
-       mutex_lock(&input_dev->mutex);
+       guard(mutex)(&input_dev->mutex);
 
        if (input_device_enabled(input_dev))
                kxtj9_enable(tj9);
 
-       mutex_unlock(&input_dev->mutex);
        return 0;
 }