diff --git a/Project_Model/Libs/BPE/Classes/NanoSocraTrainerPool.py b/Project_Model/Libs/BPE/Classes/NanoSocraTrainerPool.py
index cdd7a95..4dd4f4f 100644
--- a/Project_Model/Libs/BPE/Classes/NanoSocraTrainerPool.py
+++ b/Project_Model/Libs/BPE/Classes/NanoSocraTrainerPool.py
@@ -46,7 +46,7 @@ def split_fit(object: tuple[NanoSocratesBPE, list[list[int]]]):
             continue
 
         # We are sure of its type
-        NEW_DATA.append(output)  # type: ignore
+        NEW_DATA.append(piece)  # type: ignore
 
     return (bpe, NEW_DATA, memory)
 
@@ -56,14 +56,14 @@ def split_encode(object: tuple[NanoSocratesBPE, list[list[int]]]):
 
     NEW_DATA: list[list[int]] = []
 
-    for piece in data:
+    for index, piece in zip(range(0, len(data)), data):
         output = bpe.encode_intermediate(piece)
 
         if len(output) < 2:
             continue
 
         # We are sure of its type
-        NEW_DATA.append(output)  # type: ignore
+        NEW_DATA.append(data[index])  # type: ignore
 
     return NEW_DATA