diff --git a/drivers/mtd/devices/docg3.c b/drivers/mtd/devices/docg3.c
index f7e33dc..88d707c 100644
--- a/drivers/mtd/devices/docg3.c
+++ b/drivers/mtd/devices/docg3.c
@@ -614,7 +614,7 @@
 	return !is_good;
 }
 
-#ifdef MTD_WRITE
+#ifdef CONFIG_MTD_WRITE
 static int doc_guess_autoecc(struct mtd_oob_ops *ops)
 {
 	int autoecc;
@@ -1065,7 +1065,7 @@
 	mtd->read = doc_read;
 	mtd->read_oob = doc_read_oob;
 	mtd->block_isbad = doc_block_isbad;
-#ifdef MTD_WRITE
+#ifdef CONFIG_MTD_WRITE
 	mtd->erase = doc_erase;
 	mtd->write = doc_write;
 	mtd->write_oob = doc_write_oob;