diff --git a/src/soundevent/evaluation/affinity.py b/src/soundevent/evaluation/affinity.py
index 00511e9..6802289 100644
--- a/src/soundevent/evaluation/affinity.py
+++ b/src/soundevent/evaluation/affinity.py
@@ -6,9 +6,15 @@
 from soundevent import data
 
 if sys.version_info >= (3, 8):
-    from typing import ParamSpec, Protocol
+    from typing import Protocol
 else:
-    from typing_extensions import ParamSpec, Protocol
+    from typing_extensions import Protocol
+
+if sys.version_info >= (3, 10):
+    from typing import ParamSpec
+else:
+    from typing_extensions import ParamSpec
+
 
 P = ParamSpec("P")