diff --git a/include/chipdrivers.h b/include/chipdrivers.h
index 8e2c520..8c4ecba 100644
--- a/include/chipdrivers.h
+++ b/include/chipdrivers.h
@@ -215,4 +215,8 @@
 void decode_range_spi25_bit_cmp(size_t *start, size_t *len, const struct wp_bits *, size_t chip_len);
 void decode_range_spi25_2x_block(size_t *start, size_t *len, const struct wp_bits *, size_t chip_len);
 
+/* flashprog.c */
+int prepare_memory_access(struct flashctx *, enum preparation_steps);
+void finish_memory_access(struct flashctx *);
+
 #endif /* !__CHIPDRIVERS_H__ */
diff --git a/include/flash.h b/include/flash.h
index 7afdf32..c71b1d9 100644
--- a/include/flash.h
+++ b/include/flash.h
@@ -211,6 +211,7 @@
 struct wp_bits;
 
 enum preparation_steps {
+	PREPARE_PROBE,
 	PREPARE_FULL,
 };