diff --git a/tests/unit/encoders/test_tfidf.py b/tests/unit/encoders/test_tfidf.py
index 68e37d9e955ad93bccc93f1ea7ea591709de292c..9e4cae29a781e5c5dd30ff9d7775fc15f5a9ff4c 100644
--- a/tests/unit/encoders/test_tfidf.py
+++ b/tests/unit/encoders/test_tfidf.py
@@ -1,6 +1,6 @@
 import pytest
 from semantic_router.encoders import TfidfEncoder
-from semantic_router.schemas.route import Route
+from semantic_router.route import Route
 
 
 @pytest.fixture