diff --git a/test/reference/ddr3_init.h b/test/reference/ddr3_init.h
index d1c7d66..80b82a2 100644
--- a/test/reference/ddr3_init.h
+++ b/test/reference/ddr3_init.h
@@ -39,15 +39,15 @@ __attribute__((unused)) static void command_p3(int cmd)
 
 #define DFII_PIX_DATA_SIZE CSR_SDRAM_DFII_PI0_WRDATA_SIZE
 
-const unsigned long sdram_dfii_pix_wrdata_addr[4] = {
-    CSR_SDRAM_DFII_PI0_WRDATA_ADDR,
+const unsigned long sdram_dfii_pix_wrdata_addr[DFII_NPHASES] = {
+	CSR_SDRAM_DFII_PI0_WRDATA_ADDR,
 	CSR_SDRAM_DFII_PI1_WRDATA_ADDR,
 	CSR_SDRAM_DFII_PI2_WRDATA_ADDR,
 	CSR_SDRAM_DFII_PI3_WRDATA_ADDR
 };
 
-const unsigned long sdram_dfii_pix_rddata_addr[4] = {
-    CSR_SDRAM_DFII_PI0_RDDATA_ADDR,
+const unsigned long sdram_dfii_pix_rddata_addr[DFII_NPHASES] = {
+	CSR_SDRAM_DFII_PI0_RDDATA_ADDR,
 	CSR_SDRAM_DFII_PI1_RDDATA_ADDR,
 	CSR_SDRAM_DFII_PI2_RDDATA_ADDR,
 	CSR_SDRAM_DFII_PI3_RDDATA_ADDR
diff --git a/test/reference/ddr4_init.h b/test/reference/ddr4_init.h
index c970d97..2f99c9d 100644
--- a/test/reference/ddr4_init.h
+++ b/test/reference/ddr4_init.h
@@ -39,15 +39,15 @@ __attribute__((unused)) static void command_p3(int cmd)
 
 #define DFII_PIX_DATA_SIZE CSR_SDRAM_DFII_PI0_WRDATA_SIZE
 
-const unsigned long sdram_dfii_pix_wrdata_addr[4] = {
-    CSR_SDRAM_DFII_PI0_WRDATA_ADDR,
+const unsigned long sdram_dfii_pix_wrdata_addr[DFII_NPHASES] = {
+	CSR_SDRAM_DFII_PI0_WRDATA_ADDR,
 	CSR_SDRAM_DFII_PI1_WRDATA_ADDR,
 	CSR_SDRAM_DFII_PI2_WRDATA_ADDR,
 	CSR_SDRAM_DFII_PI3_WRDATA_ADDR
 };
 
-const unsigned long sdram_dfii_pix_rddata_addr[4] = {
-    CSR_SDRAM_DFII_PI0_RDDATA_ADDR,
+const unsigned long sdram_dfii_pix_rddata_addr[DFII_NPHASES] = {
+	CSR_SDRAM_DFII_PI0_RDDATA_ADDR,
 	CSR_SDRAM_DFII_PI1_RDDATA_ADDR,
 	CSR_SDRAM_DFII_PI2_RDDATA_ADDR,
 	CSR_SDRAM_DFII_PI3_RDDATA_ADDR
diff --git a/test/reference/sdr_init.h b/test/reference/sdr_init.h
index 9337aec..b091c7a 100644
--- a/test/reference/sdr_init.h
+++ b/test/reference/sdr_init.h
@@ -24,12 +24,12 @@ __attribute__((unused)) static void command_p0(int cmd)
 
 #define DFII_PIX_DATA_SIZE CSR_SDRAM_DFII_PI0_WRDATA_SIZE
 
-const unsigned long sdram_dfii_pix_wrdata_addr[1] = {
-    CSR_SDRAM_DFII_PI0_WRDATA_ADDR
+const unsigned long sdram_dfii_pix_wrdata_addr[DFII_NPHASES] = {
+	CSR_SDRAM_DFII_PI0_WRDATA_ADDR
 };
 
-const unsigned long sdram_dfii_pix_rddata_addr[1] = {
-    CSR_SDRAM_DFII_PI0_RDDATA_ADDR
+const unsigned long sdram_dfii_pix_rddata_addr[DFII_NPHASES] = {
+	CSR_SDRAM_DFII_PI0_RDDATA_ADDR
 };
 
 static void init_sequence(void)