diff --git a/tests/unit/test_schema.py b/tests/unit/test_schema.py
index f4f9762368a987340c95c20bf95f50dbc5c63447..97b5028e448ea4683c2df09aa93c8946447c8b28 100644
--- a/tests/unit/test_schema.py
+++ b/tests/unit/test_schema.py
@@ -1,6 +1,6 @@
 import pytest
 
-from semantic_router.schemas.encoder import (
+from semantic_router.schema import (
     CohereEncoder,
     Encoder,
     EncoderType,