diff --git a/drivers/input/touchscreen/nova_0flash_mmi/Android.mk b/drivers/input/touchscreen/nova_0flash_mmi/Android.mk
index 0d4822722811..6769516c59e7 100644
--- a/drivers/input/touchscreen/nova_0flash_mmi/Android.mk
+++ b/drivers/input/touchscreen/nova_0flash_mmi/Android.mk
@@ -17,7 +17,10 @@ endif
 
 ifneq ($(MOTO_PANEL_CHECK_TOUCH_STATE),)
 	KBUILD_OPTIONS += CONFIG_INPUT_NOVA_0FLASH_MMI_NOTIFY_TOUCH_STATE=y
-	LOCAL_ADDITIONAL_DEPENDENCIES += $(KERNEL_MODULES_OUT)/msm_drm.ko
+endif
+
+ifneq ($(MOTO_PANEL_CHECK_TOUCH_STATE_V2),)
+	KBUILD_OPTIONS += CONFIG_INPUT_NOVA_0FLASH_MMI_NOTIFY_TOUCH_STATE=y
 endif
 
 ifneq ($(BOARD_USES_PANEL_NOTIFICATIONS),)
@@ -87,7 +90,7 @@ ifneq ($(findstring touchscreen_mmi.ko,$(BOARD_VENDOR_KERNEL_MODULES)),)
 	KBUILD_OPTIONS += CONFIG_INPUT_TOUCHSCREEN_MMI=y
 	LOCAL_ADDITIONAL_DEPENDENCIES += $(KERNEL_MODULES_OUT)/touchscreen_mmi.ko
 endif
-ifneq ($(MOTO_PANEL_CHECK_TOUCH_STATE),)
+ifneq ($(MOTO_PANEL_CHECK_TOUCH_STATE_V2),)
 	LOCAL_ADDITIONAL_DEPENDENCIES += $(KERNEL_MODULES_OUT)/msm_drm.ko
 endif
 KBUILD_OPTIONS_GKI += GKI_OBJ_MODULE_DIR=gki