diff --git a/src/ptbench/config/data/tbx11k/datamodule.py b/src/ptbench/config/data/tbx11k/datamodule.py
index 7d3d786e856389e4c1a189140a71a304128a3d5e..abe2bec7d35cf451fb98cca977fa69558935613b 100644
--- a/src/ptbench/config/data/tbx11k/datamodule.py
+++ b/src/ptbench/config/data/tbx11k/datamodule.py
@@ -9,6 +9,7 @@ import os
 import typing
 
 import PIL.Image
+import typing_extensions
 
 from torch.utils.data._utils.collate import default_collate_fn_map
 from torchvision.transforms.functional import to_tensor
@@ -63,7 +64,7 @@ class BoundingBox:
     def ymax(self) -> int:
         return self.ymin + self.height - 1
 
-    def intersection(self, other: typing.Self) -> int:
+    def intersection(self, other: typing_extensions.Self) -> int:
         """Computes the area intersection between bounding boxes.
 
         Notice that screen geometry dictates is slightly different from