diff --git a/tests/unit/encoders/test_vit.py b/tests/unit/encoders/test_vit.py
index 9970e31075a44396c5fbfeaff0a2ce0a846487ca..d7e3d3bfd06a522fefb0e164f5ad519121aaf4a2 100644
--- a/tests/unit/encoders/test_vit.py
+++ b/tests/unit/encoders/test_vit.py
@@ -6,7 +6,7 @@ from PIL import Image
 from semantic_router.encoders import VitEncoder
 from transformers import AutoTokenizer, AutoModel
 
-test_model_name = "hf-internal-testing/tiny-random-vit"
+test_model_name = "aurelio-ai/sr-test-vit"
 
 # force the model download
 tokenizer = AutoTokenizer.from_pretrained(test_model_name, force_download=True)