diff --git a/hash_objects/hash_set_bst.py b/hash_objects/hash_set_bst.py
index 54b53a2..9f3b351 100644
--- a/hash_objects/hash_set_bst.py
+++ b/hash_objects/hash_set_bst.py
@@ -5,8 +5,8 @@
 
 class HashSet:
 
-    def __init__(self):
-        self.size = 131
+    def __init__(self, size):
+        self.size = size
         self.bucket = [Bucket() for _ in range(self.size)]
 
     def _get_hash_key(self, key):