diff --git a/ch347_spi.c b/ch347_spi.c
index 7772fe0..cd334da 100644
--- a/ch347_spi.c
+++ b/ch347_spi.c
@@ -176,10 +176,10 @@
 	return 0;
 }
 
-static int ch347_spi_send_command(const struct flashctx *flash, unsigned int writecnt,
+static int ch347_spi_send_command(const struct spi_master *mst, unsigned int writecnt,
 		unsigned int readcnt, const unsigned char *writearr, unsigned char *readarr)
 {
-	struct ch347_spi_data *ch347_data = flash->mst.spi->data;
+	struct ch347_spi_data *ch347_data = mst->data;
 	int ret = 0;
 
 	ch347_cs_control(ch347_data, CH347_CS_ASSERT | CH347_CS_CHANGE, CH347_CS_IGNORE);