diff --git a/hash_objects/hash_map_array.py b/hash_objects/hash_map_array.py
index 6ec0031..6508815 100644
--- a/hash_objects/hash_map_array.py
+++ b/hash_objects/hash_map_array.py
@@ -35,7 +35,7 @@ class Bucket:
 
 class HashMap:
 
-    def __init__(self, key_space):
+    def __init__(self, size):
         self.size = size
         self.table = [Bucket() for _ in range(self.size)]