diff --git a/src/axolotl/core/trainer_builder.py b/src/axolotl/core/trainer_builder.py
index c9cc2070e..325c02ef3 100755
--- a/src/axolotl/core/trainer_builder.py
+++ b/src/axolotl/core/trainer_builder.py
@@ -25,7 +25,7 @@ import os
 import sys
 from abc import abstractmethod
 from pathlib import Path
-from typing import List, Type, Union
+from typing import Type, Union
 
 import torch
 import transformers
@@ -155,7 +155,7 @@ class TrainerBuilderBase(abc.ABC):
     def build(self, total_num_steps):
         pass
 
-    def get_callbacks(self) -> List[TrainerCallback]:
+    def get_callbacks(self) -> list[TrainerCallback]:
         callbacks = []
 
         plugin_manager = PluginManager.get_instance()