diff --git a/drivers/input/touchscreen/synaptics_mmi/synaptics_dsx_i2c.c b/drivers/input/touchscreen/synaptics_mmi/synaptics_dsx_i2c.c
index 37c542c33ecf..a98b8118a5bf 100644
--- a/drivers/input/touchscreen/synaptics_mmi/synaptics_dsx_i2c.c
+++ b/drivers/input/touchscreen/synaptics_mmi/synaptics_dsx_i2c.c
@@ -36,7 +36,6 @@
 #include <linux/pinctrl/consumer.h>
 #include <linux/input/mt.h>
 #include <linux/input/synaptics_dsx_mmi.h>
-#include <linux/msm_drm_notify.h>
 
 #include "synaptics_dsx_i2c.h"
 #include "synaptics_dsx_dropbox.h"
diff --git a/drivers/input/touchscreen/synaptics_mmi/synaptics_dsx_i2c.h b/drivers/input/touchscreen/synaptics_mmi/synaptics_dsx_i2c.h
index 49e851e811e9..2797ebd92e56 100644
--- a/drivers/input/touchscreen/synaptics_mmi/synaptics_dsx_i2c.h
+++ b/drivers/input/touchscreen/synaptics_mmi/synaptics_dsx_i2c.h
@@ -35,6 +35,9 @@
 #endif
 #if defined(CONFIG_MMI_PANEL_NOTIFICATIONS)
 #include <linux/mmi_panel_notifier.h>
+#elif defined(CONFIG_DRM)
+#include <linux/notifier.h>
+#include <linux/msm_drm_notify.h>
 #elif defined(CONFIG_FB)
 #include <linux/notifier.h>
 #include <linux/fb.h>
@@ -474,7 +477,7 @@ struct synaptics_rmi4_data {
 	struct mutex state_mutex;
 #if defined(CONFIG_MMI_PANEL_NOTIFICATIONS)
 	struct mmi_notifier panel_nb;
-#elif defined(CONFIG_FB)
+#elif (defined(CONFIG_FB) || defined(CONFIG_DRM))
 	struct notifier_block panel_nb;
 #endif
 	atomic_t panel_off_flag;