diff --git a/common/spl/spl_fit.c b/common/spl/spl_fit.c
index d65cbb1ae6..7203d30d78 100644
--- a/common/spl/spl_fit.c
+++ b/common/spl/spl_fit.c
@@ -197,7 +197,7 @@ static int spl_load_fit_image(struct spl_load_info *info, ulong sector,
 			return -ENOENT;
 
 		load_ptr = (load_addr + align_len) & ~align_len;
-#if  defined(CONFIG_ROCKCHIP_RK3399) || defined(CONFIG_ROCKCHIP_RK3368)
+#if  defined(CONFIG_ARCH_ROCKCHIP)
 		if ((load_ptr & 0xffff0000) == CONFIG_ROCKCHIP_IRAM_START_ADDR)
 			load_ptr = (ulong)memalign(ARCH_DMA_MINALIGN, len);
 #endif