diff --git a/include/chipdrivers.h b/include/chipdrivers.h
index 0b1f1f6..f936b6a 100644
--- a/include/chipdrivers.h
+++ b/include/chipdrivers.h
@@ -128,7 +128,7 @@
 int erase_opaque(struct flashctx *flash, unsigned int blockaddr, unsigned int blocklen);
 
 /* at45db.c */
-int probe_spi_at45db(struct flashctx *flash);
+int spi_prepare_at45db(struct flashctx *, const enum preparation_steps);
 int spi_prettyprint_status_register_at45db(struct flashctx *flash);
 int spi_disable_blockprotect_at45db(struct flashctx *flash);
 int spi_read_at45db(struct flashctx *flash, uint8_t *buf, unsigned int start, unsigned int len);