* configuration details.
  */
 static int write_page(struct mtd_info *mtd, struct nand_chip *chip,
-                       const uint8_t *buf, bool raw_xfer)
+                       const uint8_t *buf, int page, bool raw_xfer)
 {
        struct denali_nand_info *denali = mtd_to_denali(mtd);
        dma_addr_t addr = denali->buf.dma_buf;
        size_t size = mtd->writesize + mtd->oobsize;
        uint32_t irq_status;
        uint32_t irq_mask = INTR__DMA_CMD_COMP | INTR__PROGRAM_FAIL;
+       int ret = 0;
+
+       denali->page = page;
 
        /*
         * if it is a raw xfer, we want to disable ecc and send the spare area.
        if (irq_status == 0) {
                dev_err(denali->dev, "timeout on write_page (type = %d)\n",
                        raw_xfer);
-               denali->status = NAND_STATUS_FAIL;
+               ret = -EIO;
        }
 
        denali_enable_dma(denali, false);
        dma_sync_single_for_cpu(denali->dev, addr, size, DMA_TO_DEVICE);
 
-       return 0;
+       return ret;
 }
 
 /* NAND core entry points */
         * for regular page writes, we let HW handle all the ECC
         * data written to the device.
         */
-       return write_page(mtd, chip, buf, false);
+       return write_page(mtd, chip, buf, page, false);
 }
 
 /*
         * for raw page writes, we want to disable ECC and simply write
         * whatever data is in the buffer.
         */
-       return write_page(mtd, chip, buf, true);
+       return write_page(mtd, chip, buf, page, true);
 }
 
 static int denali_write_oob(struct mtd_info *mtd, struct nand_chip *chip,
        unsigned long uncor_ecc_flags = 0;
        int stat = 0;
 
-       if (page != denali->page) {
-               dev_err(denali->dev,
-                       "IN %s: page %d is not equal to denali->page %d",
-                       __func__, page, denali->page);
-               BUG();
-       }
+       denali->page = page;
 
        setup_ecc_for_xfer(denali, true, false);
 
        size_t size = mtd->writesize + mtd->oobsize;
        uint32_t irq_mask = INTR__DMA_CMD_COMP;
 
-       if (page != denali->page) {
-               dev_err(denali->dev,
-                       "IN %s: page %d is not equal to denali->page %d",
-                       __func__, page, denali->page);
-               BUG();
-       }
+       denali->page = page;
 
        setup_ecc_for_xfer(denali, false, true);
        denali_enable_dma(denali, true);
 
 static int denali_waitfunc(struct mtd_info *mtd, struct nand_chip *chip)
 {
-       struct denali_nand_info *denali = mtd_to_denali(mtd);
-       int status = denali->status;
-
-       denali->status = 0;
-
-       return status;
+       return 0;
 }
 
 static int denali_erase(struct mtd_info *mtd, int page)
        int i;
 
        switch (cmd) {
-       case NAND_CMD_PAGEPROG:
-               break;
        case NAND_CMD_STATUS:
                read_status(denali);
                break;
                        write_byte_to_buf(denali, id);
                }
                break;
-       case NAND_CMD_READ0:
-       case NAND_CMD_SEQIN:
-               denali->page = page;
-               break;
        case NAND_CMD_RESET:
                reset_bank(denali);
                break;
 
        mtd_set_ooblayout(mtd, &denali_ooblayout_ops);
 
+       chip->ecc.options |= NAND_ECC_CUSTOM_PAGE_ACCESS;
        chip->ecc.read_page = denali_read_page;
        chip->ecc.read_page_raw = denali_read_page_raw;
        chip->ecc.write_page = denali_write_page;