diff --git a/drivers/memory/spiflash.c b/drivers/memory/spiflash.c
index ea0fef8..ad451f2 100644
--- a/drivers/memory/spiflash.c
+++ b/drivers/memory/spiflash.c
@@ -162,7 +162,7 @@
     if (self->config->bus_kind == MP_SPIFLASH_BUS_SPI) {
         mp_hal_pin_write(self->config->bus.u_spi.cs, 1);
         mp_hal_pin_output(self->config->bus.u_spi.cs);
-        self->config->bus.u_spi.proto->init(self->config->bus.u_spi.data, 0, NULL, (mp_map_t*)&mp_const_empty_map);
+        self->config->bus.u_spi.proto->ioctl(self->config->bus.u_spi.data, MP_SPI_IOCTL_INIT);
     } else {
         self->config->bus.u_qspi.proto->ioctl(self->config->bus.u_qspi.data, MP_QSPI_IOCTL_INIT);
     }
diff --git a/drivers/memory/spiflash.h b/drivers/memory/spiflash.h
index 79ba549..03bad52 100644
--- a/drivers/memory/spiflash.h
+++ b/drivers/memory/spiflash.h
@@ -26,8 +26,8 @@
 #ifndef MICROPY_INCLUDED_DRIVERS_MEMORY_SPIFLASH_H
 #define MICROPY_INCLUDED_DRIVERS_MEMORY_SPIFLASH_H
 
+#include "drivers/bus/spi.h"
 #include "drivers/bus/qspi.h"
-#include "extmod/machine_spi.h"
 
 enum {
     MP_SPIFLASH_BUS_SPI,
@@ -40,7 +40,7 @@
         struct {
             mp_hal_pin_obj_t cs;
             void *data;
-            const mp_machine_spi_p_t *proto;
+            const mp_spi_proto_t *proto;
         } u_spi;
         struct {
             void *data;