diff --git a/face_swapper/src/types.py b/face_swapper/src/types.py
index 3190341..74ba33e 100644
--- a/face_swapper/src/types.py
+++ b/face_swapper/src/types.py
@@ -1,6 +1,5 @@
-from typing import Any, Dict, Tuple, TypeAlias
+from typing import Any, Dict, Tuple, TypeAlias, Type, Literal
 
-from jinja2.nodes import Literal
 from torch import Tensor
 from torch.nn import Module