diff --git a/ni845x_spi.c b/ni845x_spi.c
index 177c799..21d39a8 100644
--- a/ni845x_spi.c
+++ b/ni845x_spi.c
@@ -567,7 +567,7 @@
 	return 0;
 }
 
-static int ni845x_spi_transmit(const struct flashctx *flash,
+static int ni845x_spi_transmit(const struct spi_master *mst,
 			       unsigned int write_cnt,
 			       unsigned int read_cnt,
 			       const unsigned char *write_arr,