diff --git a/python/python/tests/test_pydantic.py b/python/python/tests/test_pydantic.py
index 97c8a0aa..d8d57e2e 100644
--- a/python/python/tests/test_pydantic.py
+++ b/python/python/tests/test_pydantic.py
@@ -30,6 +30,15 @@ from lancedb.pydantic import (
 )
 from pydantic import Field
 
+from lancedb.pydantic import (
+    PYDANTIC_VERSION,
+    LanceModel,
+    Vector,
+    pydantic_to_schema,
+    EncodedImage,
+    ImageURI,
+)
+
 
 @pytest.mark.skipif(
     sys.version_info < (3, 9),