diff --git a/drivers/input/touchscreen/nova_0flash_mmi/nt36xxx_mmi_class.c b/drivers/input/touchscreen/nova_0flash_mmi/nt36xxx_mmi_class.c
index ea9f56bdc7b5..74d2b2b8fc66 100644
--- a/drivers/input/touchscreen/nova_0flash_mmi/nt36xxx_mmi_class.c
+++ b/drivers/input/touchscreen/nova_0flash_mmi/nt36xxx_mmi_class.c
@@ -536,6 +536,12 @@ static int nvt_mmi_firmware_update(struct device *dev, char *fwname)
 	nvt_update_firmware(nvt_boot_firmware_name);
 	mutex_unlock(&ts->lock);
 
+#ifdef NOVATECH_PEN_NOTIFIER
+	if(!ts->fw_ready_flag)
+		ts->fw_ready_flag = true;
+	nvt_mcu_pen_detect_set(ts->nvt_pen_detect_flag);
+#endif
+
 	return 0;
 }