Update matrix_test.go