diff --git a/Project_Model/Tests/bpe_trainer_test.py b/Project_Model/Tests/bpe_trainer_test.py
index 69ac4bb..1f9f7fb 100644
--- a/Project_Model/Tests/bpe_trainer_test.py
+++ b/Project_Model/Tests/bpe_trainer_test.py
@@ -12,8 +12,7 @@ class TestTrainBPE:
 
         TRAINER = BPE.NanoSocraTrainerPool(
             int(32E3),
-            ["<SOT>", "<EOT>"],
-            40
+            ["<SOT>", "<EOT>"]
         )
 
         TEXT = "abababab"