diff --git a/hercules_cg/src/gpu.rs b/hercules_cg/src/gpu.rs
index 01ca4773907c19fd536c0eb3034095baffbb7309..c093949f5f54ed5ac2475c6f306e4114a2b10b41 100644
--- a/hercules_cg/src/gpu.rs
+++ b/hercules_cg/src/gpu.rs
@@ -2008,7 +2008,11 @@ extern \"C\" {} {}(",
                 } else {
                     multiply_dcs(extents)
                 };
-                format!("{} * {}", self.get_alignment(*element_type), array_size)
+                format!(
+                    "{} * {}",
+                    self.get_size(*element_type, None, extra_dim_collects),
+                    array_size
+                )
             }
             Type::Product(fields) => {
                 let num_fields = num_fields.unwrap_or(fields.len());