diff --git a/at45db.c b/at45db.c
index f84f3f4..2768ace 100644
--- a/at45db.c
+++ b/at45db.c
@@ -172,14 +172,11 @@
 }
 
 /* Adapt chip entry for AT45DB* chips that support multiple page sizes. */
-int spi_prepare_at45db(struct flashctx *const flash, const enum preparation_steps prep)
+int spi_prepare_at45db(struct flashctx *const flash)
 {
 	struct flashchip *const chip = flash->chip;
 	uint8_t status;
 
-	if (prep != PREPARE_POST_PROBE)
-		return 0;
-
 	/* Power-of-2 check */
 	if (chip->page_size & (chip->page_size - 1))
 		return 0;