diff --git a/python/python/tests/test_pydantic.py b/python/python/tests/test_pydantic.py
index 3cbc7dcc..a23bc166 100644
--- a/python/python/tests/test_pydantic.py
+++ b/python/python/tests/test_pydantic.py
@@ -23,6 +23,7 @@ import pyarrow as pa
 import pydantic
 import pytest
 import pytz
+from lance.arrow import ImageURIArray
 from pydantic import Field
 
 from lancedb.pydantic import (