diff --git a/src/pyrite/rtl/pyrite_pcie_us_vpd_bpi.sv b/src/pyrite/rtl/pyrite_pcie_us_vpd_bpi.sv
index b1de028..a348326 100644
--- a/src/pyrite/rtl/pyrite_pcie_us_vpd_bpi.sv
+++ b/src/pyrite/rtl/pyrite_pcie_us_vpd_bpi.sv
@@ -241,10 +241,10 @@ always_ff @(posedge clk) begin
             8'h48: vpd_apb_prdata_reg <= 0;               // BPI flash ctrl: Next header
             8'h4C: begin
                 // BPI flash ctrl: format
-                vpd_apb_prdata_reg[3:0]   <= FLASH_SEG_COUNT;        // configuration (two segments)
-                vpd_apb_prdata_reg[7:4]   <= FLASH_SEG_DEFAULT;      // default segment
-                vpd_apb_prdata_reg[11:8]  <= FLASH_SEG_FALLBACK;     // fallback segment
-                vpd_apb_prdata_reg[31:12] <= FLASH_SEG0_SIZE >> 12;  // first segment size (even split)
+                vpd_apb_prdata_reg[3:0]   <= FLASH_SEG_COUNT;     // configuration
+                vpd_apb_prdata_reg[7:4]   <= FLASH_SEG_DEFAULT;   // default segment
+                vpd_apb_prdata_reg[11:8]  <= FLASH_SEG_FALLBACK;  // fallback segment
+                vpd_apb_prdata_reg[31:12] <= 20'(FLASH_SEG0_SIZE >> 12);  // first segment size
             end
             8'h50: begin
                 // BPI flash ctrl: control
diff --git a/src/pyrite/rtl/pyrite_pcie_us_vpd_qspi.sv b/src/pyrite/rtl/pyrite_pcie_us_vpd_qspi.sv
index d7b8a2d..60ac1d1 100644
--- a/src/pyrite/rtl/pyrite_pcie_us_vpd_qspi.sv
+++ b/src/pyrite/rtl/pyrite_pcie_us_vpd_qspi.sv
@@ -243,10 +243,10 @@ always_ff @(posedge clk) begin
             8'h48: vpd_apb_prdata_reg <= 0;               // SPI flash ctrl: Next header
             8'h4C: begin
                 // SPI flash ctrl: format
-                vpd_apb_prdata_reg[3:0]   <= FLASH_SEG_COUNT;        // configuration (two segments)
-                vpd_apb_prdata_reg[7:4]   <= FLASH_SEG_DEFAULT;      // default segment
-                vpd_apb_prdata_reg[11:8]  <= FLASH_SEG_FALLBACK;     // fallback segment
-                vpd_apb_prdata_reg[31:12] <= FLASH_SEG0_SIZE >> 12;  // first segment size (even split)
+                vpd_apb_prdata_reg[3:0]   <= FLASH_SEG_COUNT;     // configuration
+                vpd_apb_prdata_reg[7:4]   <= FLASH_SEG_DEFAULT;   // default segment
+                vpd_apb_prdata_reg[11:8]  <= FLASH_SEG_FALLBACK;  // fallback segment
+                vpd_apb_prdata_reg[31:12] <= 20'(FLASH_SEG0_SIZE >> 12);  // first segment size
             end
             8'h50: begin
                 // SPI flash ctrl: control 0