diff --git a/atomlib/atoms.py b/atomlib/atoms.py
index 917bca7..ef29651 100644
--- a/atomlib/atoms.py
+++ b/atomlib/atoms.py
@@ -24,7 +24,7 @@
 import polars.testing
 import polars.type_aliases
 
-from .types import to_vec3, VecLike
+from .types import to_vec3, VecLike, ParamSpec
 from .bbox import BBox3D
 from .elem import get_elem, get_sym, get_mass
 from .transform import Transform3D, IntoTransform3D, AffineTransform3D
@@ -144,7 +144,7 @@ def _select_schema(df: t.Union[polars.DataFrame, HasAtoms], schema: SchemaDict)
 
 
 HasAtomsT = t.TypeVar('HasAtomsT', bound='HasAtoms')
-P = t.ParamSpec('P')
+P = ParamSpec('P')
 T = t.TypeVar('T')