diff --git a/src/soc/ti/am335x/cbmem.c b/src/soc/ti/am335x/cbmem.c
index 3765874..170695e 100644
--- a/src/soc/ti/am335x/cbmem.c
+++ b/src/soc/ti/am335x/cbmem.c
@@ -1,9 +1,10 @@
 /* SPDX-License-Identifier: GPL-2.0-only */
 
 #include <cbmem.h>
+#include <commonlib/bsd/helpers.h>
 #include <symbols.h>
 
 void *cbmem_top_chipset(void)
 {
-	return _dram + (CONFIG_DRAM_SIZE_MB << 20);
+	return _dram + CONFIG_DRAM_SIZE_MB * MiB;
 }