diff --git a/slangpy/slang/staticarray.slang b/slangpy/slang/staticarray.slang
index 041c81ffa..1348159f4 100644
--- a/slangpy/slang/staticarray.slang
+++ b/slangpy/slang/staticarray.slang
@@ -7,4 +7,4 @@ public interface ISizedArray<T, let D : int> { __subscript(int index)->T { get;
 // Array, vector and matrix conform to ISizedArray
 public extension<T, let D : int> Array<T, D> : ISizedArray<T, D> {}
 public extension<T, let D : int> vector<T, D> : ISizedArray<T, D> {}
-public extension<T, let R : int, let C : int, let L : int> matrix<T, R, C, L> : ISizedArray<vector<T, C>, R> {}
+public extension<T, let R : int, let C : int, let L : MatrixLayoutMode> matrix<T, R, C, L> : ISizedArray<vector<T, C>, R> {}
diff --git a/src/sgl/device/print.slang b/src/sgl/device/print.slang
index 6d2d4830e..c00b15079 100644
--- a/src/sgl/device/print.slang
+++ b/src/sgl/device/print.slang
@@ -257,7 +257,7 @@ extension vector<T, N> : IPrintable {
     uint get_printable_data(uint index) { return this[index / T::printable_data_count()].get_printable_data(index % T::printable_data_count()); }
 };
 
-__generic<T : IPrintable, let R : int, let C : int, let L : int>
+__generic<T : IPrintable, let R : int, let C : int, let L : MatrixLayoutMode>
 extension matrix<T, R, C, L> : IPrintable {
     [ForceInline]
     static Layout printable_layout() { return Layout(Kind::matrix, T::printable_layout().type, R, C); }