diff --git a/Project_Model/Tests/bpe_trainer_test.py b/Project_Model/Tests/bpe_trainer_test.py
index 2e1fa08..69ac4bb 100644
--- a/Project_Model/Tests/bpe_trainer_test.py
+++ b/Project_Model/Tests/bpe_trainer_test.py
@@ -4,13 +4,13 @@ import Project_Model.Libs.BPE as BPE
 
 import re
 
-CACHE_DIR_PATH = Path("Project_Model/Tests/trainer_files/cache")
+CACHE_DIR_PATH = Path("Project_Model/Tests/trainer_files/cache/pool-cache.json")
 
 class TestTrainBPE:
 
     def test_bpe_train_encoding_simple(self):
 
-        TRAINER = BPE.NanoSocraTrainer(
+        TRAINER = BPE.NanoSocraTrainerPool(
             int(32E3),
             ["<SOT>", "<EOT>"],
             40