diff --git a/tests/unit/test_cotiled_diag.py b/tests/unit/test_cotiled_diag.py
index 199a274d..47bc8e47 100644
--- a/tests/unit/test_cotiled_diag.py
+++ b/tests/unit/test_cotiled_diag.py
@@ -44,7 +44,7 @@ def main():
 
     # V FMHA layout: use explicit constants
     # pv_n_tile=256, s_k=128
-    v_major = LayoutEnum.COLUMN_MAJOR  # layout (256, 128, 1) stride (1, 256, 32768) = col-major
+    v_major = LayoutEnum.COL_MAJOR  # layout (256, 128, 1) stride (1, 256, 32768) = col-major
 
     c_layout = LayoutEnum.from_tensor(mC)