}
 #endif
 
-#define        HW_DIGCTRL_SCRATCH0     0x8001c280
-#define        HW_DIGCTRL_SCRATCH1     0x8001c290
 int mx28_dram_init(void)
 {
+       struct mx28_digctl_regs *digctl_regs =
+               (struct mx28_digctl_regs *)MXS_DIGCTL_BASE;
        uint32_t sz[2];
 
-       sz[0] = readl(HW_DIGCTRL_SCRATCH0);
-       sz[1] = readl(HW_DIGCTRL_SCRATCH1);
+       sz[0] = readl(&digctl_regs->hw_digctl_scratch0);
+       sz[1] = readl(&digctl_regs->hw_digctl_scratch1);
 
        if (sz[0] != sz[1]) {
                printf("MX28:\n"
 
                &power_regs->hw_power_vdddctrl);
 }
 
-#define        HW_DIGCTRL_SCRATCH0     0x8001c280
-#define        HW_DIGCTRL_SCRATCH1     0x8001c290
 void data_abort_memdetect_handler(void) __attribute__((naked));
 void data_abort_memdetect_handler(void)
 {
 
 void mx28_mem_get_size(void)
 {
+       struct mx28_digctl_regs *digctl_regs =
+               (struct mx28_digctl_regs *)MXS_DIGCTL_BASE;
        uint32_t sz, da;
        uint32_t *vt = (uint32_t *)0x20;
 
        vt[4] = (uint32_t)&data_abort_memdetect_handler;
 
        sz = get_ram_size((long *)PHYS_SDRAM_1, PHYS_SDRAM_1_SIZE);
-       writel(sz, HW_DIGCTRL_SCRATCH0);
-       writel(sz, HW_DIGCTRL_SCRATCH1);
+       writel(sz, &digctl_regs->hw_digctl_scratch0);
+       writel(sz, &digctl_regs->hw_digctl_scratch1);
 
        /* Restore the old DABT handler. */
        vt[4] = da;