diff --git a/components/spi_flash/esp_flash_api.c b/components/spi_flash/esp_flash_api.c
index a3a932cf0e..81f36ae945 100644
--- a/components/spi_flash/esp_flash_api.c
+++ b/components/spi_flash/esp_flash_api.c
@@ -1130,11 +1130,13 @@ esp_err_t IRAM_ATTR esp_flash_write_encrypted(esp_flash_t *chip, uint32_t addres
 #endif //CONFIG_SPI_FLASH_VERIFY_WRITE
 
     esp_err_t err = rom_spiflash_api_funcs->chip_check(&chip);
+    VERIFY_CHIP_OP(write);
     // Flash encryption only support on main flash.
     if (chip != esp_flash_default_chip) {
         return ESP_ERR_NOT_SUPPORTED;
     }
-    if (err != ESP_OK) return err;
+    CHECK_WRITE_ADDRESS(chip, address, length);
+
     if (buffer == NULL || address + length > chip->size) {
         return ESP_ERR_INVALID_ARG;
     }