diff --git a/video_transformers/trainer.py b/video_transformers/trainer.py
index e78bc30..780533b 100644
--- a/video_transformers/trainer.py
+++ b/video_transformers/trainer.py
@@ -1,5 +1,5 @@
 from pathlib import Path
-from typing import Dict, Union
+from typing import Dict, List, Union
 
 import numpy as np
 import torch
@@ -30,7 +30,7 @@ def __init__(
         mixed_precision: str = "no",
         output_dir: str = "runs",
         seed: int = 42,
-        trackers: list[GeneralTracker] = None,
+        trackers: List[GeneralTracker] = None,
         checkpoint_save: bool = True,
         checkpoint_save_interval: int = 1,
         checkpoint_save_policy: str = "epoch",