diff --git a/pyproject.toml b/pyproject.toml
index 4d9ac13..c2e7105 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -77,7 +77,7 @@ include = '\.pyi?$'
 
 [tool.pytest.ini_options]
 minversion = "6.0"
-addopts = "--verbose --color yes --disable-warnings"
+addopts = "--verbose --color yes"
 testpaths = ["tests"]
 
 [tool.ruff.pycodestyle]
diff --git a/safe/trainer/collator.py b/safe/trainer/collator.py
index f5b23a5..05a67b1 100644
--- a/safe/trainer/collator.py
+++ b/safe/trainer/collator.py
@@ -12,7 +12,7 @@
 from transformers.data.data_collator import _torch_collate_batch
 from safe.tokenizer import SAFETokenizer
 
-from tokenizer import Tokenizer
+from tokenizers import Tokenizer
 
 
 class SAFECollator: