diff --git a/src/mednet/libs/classification/data/typing.py b/src/mednet/libs/classification/data/typing.py
index 8cd214053c02cc975bfc1d48c4f5418bd8c28bf5..b504f8466e4633a39831f46fc9de108ada035d8b 100644
--- a/src/mednet/libs/classification/data/typing.py
+++ b/src/mednet/libs/classification/data/typing.py
@@ -9,7 +9,7 @@ class ClassificationRawDataLoader(RawDataLoader):
     def __init__(self):
         super().__init__()
 
-    def sample(self, _: typing.Any) -> Sample:
+    def sample(self, _: tuple[str, int, typing.Any | None]) -> Sample:
         """Load whole samples from media.
 
         Parameters
diff --git a/src/mednet/libs/segmentation/data/typing.py b/src/mednet/libs/segmentation/data/typing.py
index e1d95cdbb276c58bd8e21467948c44a6343e00e2..b18b712f504897822eec4ae88b8a1e80a6c8dc32 100644
--- a/src/mednet/libs/segmentation/data/typing.py
+++ b/src/mednet/libs/segmentation/data/typing.py
@@ -1,5 +1,3 @@
-import typing
-
 from mednet.libs.common.data.typing import RawDataLoader, Sample
 
 
@@ -9,7 +7,7 @@ class SegmentationRawDataLoader(RawDataLoader):
     def __init__(self):
         super().__init__()
 
-    def sample(self, _: typing.Any) -> Sample:
+    def sample(self, _: tuple[str, str, str | None]) -> Sample:
         """Load whole samples from media.
 
         Parameters