diff --git a/dataset.py b/dataset.py
index 1ac29ba..6c79277 100644
--- a/dataset.py
+++ b/dataset.py
@@ -31,7 +31,7 @@ class DatasetEntry:
 
 @dataclass
 class Dataset:
-    entries: List = field(default_factory=list)
+    entries: List[DatasetEntry] = field(default_factory=list)
 
     def __len__(self) -> int:
         return len(self.entries)