diff --git a/brutils/voter_id.py b/brutils/voter_id.py
index d2be3d4..db999cd 100644
--- a/brutils/voter_id.py
+++ b/brutils/voter_id.py
@@ -214,7 +214,7 @@ def _calculate_vd2(federative_union, vd1):  # type: (str, int) -> str
     return vd2
 
 
-def generate(state="ZZ") -> str:
+def generate(federative_union ="ZZ") -> str:
     """
     Generates a random valid Brazilian voter registration.
 
@@ -259,7 +259,7 @@ def generate(state="ZZ") -> str:
     if federative_union in (UFs):
         sequential_number = str(randint(0, 99999999)).zfill(8)
         uf_number = UFs[federative_union]
-        if _is_uf_number_valid(uf_number):
+        if _is_federative_union_valid(uf_number):
             vd1 = _calculate_vd1(sequential_number, uf_number)
             vd2 = _calculate_vd2(uf_number, vd1)
             return f"{sequential_number}{uf_number}{vd1}{vd2}"
diff --git a/tests/test_voter_id.py b/tests/test_voter_id.py
index ed7be32..5fabd73 100644
--- a/tests/test_voter_id.py
+++ b/tests/test_voter_id.py
@@ -92,11 +92,11 @@ def test_calculate_vd2(self):
 
     def test_generate_voter_id(self):
         # test if is_valid a voter id from MG
-        voter_id = generate(state="MG")
+        voter_id = generate(federative_union ="MG")
         self.assertIs(is_valid(voter_id), True)
 
         # test if is_valid a voter id from AC
-        voter_id = generate(state="AC")
+        voter_id = generate(federative_union ="AC")
         self.assertIs(is_valid(voter_id), True)
 
         # test if is_valid a voter id from foreigner
@@ -104,7 +104,7 @@ def test_generate_voter_id(self):
         self.assertIs(is_valid(voter_id), True)
 
         # test if UF is not valid
-        voter_id = generate(state="XX")
+        voter_id = generate(federative_union ="XX")
         self.assertIs(is_valid(voter_id), False)