diff --git a/geomagio/api/data/elements.py b/geomagio/api/data/elements.py
index 80e4aa4865a08d2deb5168ef3d600ca35d838904..67bb747f06f039efabdb719f3b0a463fddc68840 100644
--- a/geomagio/api/data/elements.py
+++ b/geomagio/api/data/elements.py
@@ -1,3 +1,6 @@
+from pydantic import BaseModel
+
+
 class Element(BaseModel):
     id: str
     abbreviation: str