diff --git a/python/tests/test_ld_matrix.py b/python/tests/test_ld_matrix.py
index ddd61aed94..dbc3695492 100644
--- a/python/tests/test_ld_matrix.py
+++ b/python/tests/test_ld_matrix.py
@@ -645,8 +645,8 @@ def two_branch_count_stat(
                       ancestral state.
     :returns: 3D array of results, dimensions (sample_sets, row_sites, col_sites).
     """
-    state_dim = len(sample_set_sizes)
-    params = {"sample_set_sizes": sample_set_sizes}
+    # state_dim = len(sample_set_sizes)
+    # params = {"sample_set_sizes": sample_set_sizes}
     result = np.zeros(
         (num_sample_sets, len(row_sites), len(col_sites)), dtype=np.float64
     )
@@ -1556,7 +1556,8 @@ def naive_matrix(ts, stat_func):
             "bottleneck_n=100_mutated",
             "n=100_m=32_rho=0.1",
             "n=100_m=32_rho=0.5",
-            # These ones fail for our naive prototype because some samples do not have a mrca
+            # These ones fail for our naive prototype because
+            # some samples do not have a mrca
             "all_fields",
             "decapitate",
             "decapitate_recomb",
@@ -1571,9 +1572,9 @@ def naive_matrix(ts, stat_func):
 )
 @pytest.mark.parametrize(
     "stat,stat_func",
-        zip(
-            ["d2_unbiased", "dz_unbiased", "pi2_unbiased"],
-            [compute_D2, compute_Dz, compute_pi2],
+    zip(
+        ["d2_unbiased", "dz_unbiased", "pi2_unbiased"],
+        [compute_D2, compute_Dz, compute_pi2],
     ),
 )
 def test_branch_ld_matrix(ts, stat, stat_func):