Minor fix to matrix size