diff --git a/README.md b/README.md
index b12ba2f..ad74868 100644
--- a/README.md
+++ b/README.md
@@ -3,4 +3,7 @@ Python implementation of Deep Learning book
 
 http://www.deeplearningbook.org/
 
-Ian Goodfellow, Yoshua Bengio and Aaron Courville, MIT Press, 2016
\ No newline at end of file
+Ian Goodfellow, Yoshua Bengio and Aaron Courville, MIT Press, 2016
+
+## Running tests
+`$ python -m unittest discover`
diff --git a/test_linear_algebra.py b/test_linear_algebra.py
index 38f2498..29a22b6 100644
--- a/test_linear_algebra.py
+++ b/test_linear_algebra.py
@@ -4,9 +4,9 @@
 
 class TestLinearAlgebra(TestCase):
     def test_transpose(self):
-        A = [[1, 2], [3, 4]]
+        A = [[1, 2, 3], [4, 5, 6]]
         actual = transpose(A)
-        expected = [[1, 3], [2, 4]]
+        expected = [[1, 4], [2, 5], [3, 6]]
         self.assertEqual(actual, expected)
 
     def test_plus(self):