diff --git a/compute.py b/compute.py
index 304353b..8d486ff 100644
--- a/compute.py
+++ b/compute.py
@@ -92,10 +92,10 @@ if __name__ == "__main__":
             "Expected a ValueError for incompatible dimensions."
         )
 
-    assert transpose_matrix(first) == [
+    assert transpose(first) == [
         [1, 4],
         [2, 5],
         [3, 6]
     ]
-    print("Transpose:", transpose_matrix(first))
+    print("Transpose:", transpose(first))