diff --git a/speeches/speechset.py b/speeches/speechset.py
index b6d7d15..4c68b5e 100644
--- a/speeches/speechset.py
+++ b/speeches/speechset.py
@@ -47,7 +47,7 @@ def split(self, size: int):
         self.dataset = self.dataset[:size]
         return residual
 
-    def __getitem__(self, index: Union[int, slice, List[int]]) -> Any:
+    def __getitem__(self, index: Union[int, slice]) -> Any:
         """Lazy normalizing.
         Args:
             index: input index.