diff --git a/arch/arm/mach-imx/esdctl.c b/arch/arm/mach-imx/esdctl.c
index bb8fec2..811592f 100644
--- a/arch/arm/mach-imx/esdctl.c
+++ b/arch/arm/mach-imx/esdctl.c
@@ -549,9 +549,9 @@
 	unsigned long base, size;
 
 	upper_or_coalesced_range(MX53_CSD0_BASE_ADDR,
-			imx_v3_sdram_size((void *)MX53_ESDCTL_BASE_ADDR, 0),
+			imx_v4_sdram_size((void *)MX53_ESDCTL_BASE_ADDR, 0),
 			MX53_CSD1_BASE_ADDR,
-			imx_v3_sdram_size((void *)MX53_ESDCTL_BASE_ADDR, 1),
+			imx_v4_sdram_size((void *)MX53_ESDCTL_BASE_ADDR, 1),
 			&base, &size);
 
 	barebox_arm_entry(base, size, boarddata);