diff --git a/src/matrix.rs b/src/matrix.rs
index a164906..20ae652 100644
--- a/src/matrix.rs
+++ b/src/matrix.rs
@@ -168,15 +168,9 @@ mod tests {
         let lattice = Matrix::new_lattice(noise_bits, ciphertexts).unwrap();
 
         let expected = vec![
-            int!(8),
-            int!(8),
-            int!(12),
-            int!(0),
-            int!(5),
-            int!(0),
-            int!(0),
-            int!(0),
-            int!(5),
+            int!(8), int!(0), int!(0),
+            int!(8), int!(5), int!(0),
+            int!(12), int!(0), int!(5),
         ];
         let mut actual = Vec::new();
         for col in 0..3 {
@@ -195,7 +189,7 @@ mod tests {
 
         let two_pow = Rational::from(Integer::from(2u64).pow(3));
         assert_eq!(lattice[(0, 0)], two_pow);
-        assert_eq!(lattice[(0, 1)], rational!(8));
-        assert_eq!(lattice[(0, 2)], rational!(12));
+        assert_eq!(lattice[(1, 0)], rational!(8));
+        assert_eq!(lattice[(2, 0)], rational!(12));
     }
 }