diff --git a/tests/test_flows.py b/tests/test_flows.py
index 6bde1c1..9b02222 100644
--- a/tests/test_flows.py
+++ b/tests/test_flows.py
@@ -81,7 +81,7 @@ def decorator(func):
 class TestS3Helper(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
-        cls.model_name = "jan-hq-test/tokenizer-tinyllama"
+        cls.model_name = "jan-hq-test/tinyllama-v1.1"
         cls.dataset_name = "jan-hq-test/test-dataset"
         S3Helper()