diff --git a/drivers/flash/soc_flash_rv32m1.c b/drivers/flash/soc_flash_rv32m1.c
index b2b41ac966d..38215f591ca 100644
--- a/drivers/flash/soc_flash_rv32m1.c
+++ b/drivers/flash/soc_flash_rv32m1.c
@@ -18,8 +18,6 @@
 #include "fsl_common.h"
 #include "fsl_flash.h"
 
-#define CONFIG_FLASH_SIZE DT_FLASH_SIZE
-
 struct flash_priv {
 	flash_config_t config;
 	/*
@@ -117,7 +115,7 @@ static int flash_mcux_write_protection(struct device *dev, bool enable)
 
 #if defined(CONFIG_FLASH_PAGE_LAYOUT)
 static const struct flash_pages_layout dev_layout = {
-	.pages_count = KB(CONFIG_FLASH_SIZE) /
+	.pages_count = DT_INST_REG_SIZE(0) /
 					DT_INST_PROP(0, erase_block_size),
 	.pages_size = DT_INST_PROP(0, erase_block_size),
 };