diff --git a/drivers/leds/leds-qti-flash.c b/drivers/leds/leds-qti-flash.c
index 321f357ef378..551818fa2495 100644
--- a/drivers/leds/leds-qti-flash.c
+++ b/drivers/leds/leds-qti-flash.c
@@ -952,6 +952,11 @@ int qti_flash_led_prepare(struct led_trigger *trig, int options,
 	snode = container_of(led_cdev, struct flash_switch_data, cdev);
 
 	if (options & QUERY_MAX_AVAIL_CURRENT) {
+		if (!max_current) {
+			pr_err("Invalid max_current pointer\n");
+			return -EINVAL;
+		}
+
 		rc = qti_flash_led_get_max_avail_current(snode->led,
 					max_current);
 		if (rc < 0) {
diff --git a/include/linux/leds-qti-flash.h b/include/linux/leds-qti-flash.h
index 898f30fde893..337ba0369f91 100644
--- a/include/linux/leds-qti-flash.h
+++ b/include/linux/leds-qti-flash.h
@@ -8,7 +8,9 @@
 
 #include <linux/leds.h>
 
-#define QUERY_MAX_AVAIL_CURRENT		BIT(0)
+#define ENABLE_REGULATOR		BIT(0)
+#define DISABLE_REGULATOR		BIT(1)
+#define QUERY_MAX_AVAIL_CURRENT		BIT(2)
 
 /**
  * struct flash_led_param: QTI flash LED parameter data