diff --git a/tests/test_ballot_generator.py b/tests/test_ballot_generator.py
index dc0f2bfc..2b3c590a 100644
--- a/tests/test_ballot_generator.py
+++ b/tests/test_ballot_generator.py
@@ -444,7 +444,7 @@ def group_elements_by_mapping(element_list, mapping):
         bloc_voter_prop=bloc_voter_prop,
         slate_to_candidates=slate_to_candidate,
         cohesion_parameters=cohesion_parameters,
-    ).generate_profile(number_of_ballots)
+    ).generate_profile(number_of_ballots=number_of_ballots)
 
     # Test
     assert do_ballot_probs_match_ballot_dist(