1 files changed, 2 insertions, 8 deletions
diff --git a/tests/test_pyhegp.py b/tests/test_pyhegp.py
index 2d2a258..c494d30 100644
--- a/tests/test_pyhegp.py
+++ b/tests/test_pyhegp.py
@@ -53,17 +53,11 @@ def no_column_zero_standard_deviation(matrix):
            elements=st.integers(min_value=0, max_value=100)),
     arrays("float64",
            array_shapes(min_dims=2, max_dims=2, min_side=2),
-           elements=st.floats(min_value=0, max_value=100)))
-       # Reject matrices with zero standard deviation columns since
-       # they trigger a division by zero.
-       .filter(no_column_zero_standard_deviation))
+           elements=st.floats(min_value=0, max_value=100))))
 def test_hegp_encryption_decryption_are_inverses(plaintext):
-    mean = np.mean(plaintext, axis=0)
-    standard_deviation = np.std(plaintext, axis=0)
     rng = np.random.default_rng()
     key = random_key(rng, len(plaintext))
-    assert hegp_decrypt(hegp_encrypt(plaintext, mean, standard_deviation, key),
-                        mean, standard_deviation, key) == approx(plaintext)
+    assert hegp_decrypt(hegp_encrypt(plaintext, key), key) == approx(plaintext)
 
 @given(arrays("float64",
               array_shapes(min_dims=2, max_dims=2),