diff --git a/python/cutlass/library_defaults.py b/python/cutlass/library_defaults.py
index f72ca394..997d5d35 100644
--- a/python/cutlass/library_defaults.py
+++ b/python/cutlass/library_defaults.py
@@ -128,7 +128,7 @@ class KernelsForDataType:
         :rtype: int
         """
         # Determine the leading dimension of the shape
-        if layout == cutlass.LayoutType.RowMajor:
+        if layout == cutlass.LayoutType.ColumnMajor:
             ld = shape[0]
         elif layout == cutlass.LayoutType.RowMajor:
             ld = shape[1]