diff --git a/edi.c b/edi.c
index 0f83a40..efe41c5 100644
--- a/edi.c
+++ b/edi.c
@@ -470,7 +470,7 @@
 	return 0;
 }
 
-static void edi_finish(struct flashctx *flash)
+void edi_finish(struct flashctx *flash)
 {
 	const struct spi_master *const spi = flash->mst.spi;
 
@@ -502,19 +502,15 @@
 	return edi_chip_probe(spi);
 }
 
-int edi_prepare(struct flashctx *flash, enum preparation_steps step)
+int edi_prepare(struct flashctx *flash)
 {
 	int rc;
 
-	if (step < PREPARE_FULL)
-		return 0;
-
 	rc = edi_8051_reset(flash->mst.spi);
 	if (rc < 0) {
 		msg_perr("%s: Unable to reset 8051!\n", __func__);
 		return rc;
 	}
 
-	flash->chip->finish_access = edi_finish;
 	return 0;
 }