diff --git a/face_swapper/src/typing.py b/face_swapper/src/typing.py
index d72df42..0154b09 100644
--- a/face_swapper/src/typing.py
+++ b/face_swapper/src/typing.py
@@ -17,6 +17,6 @@ Padding = Tuple[int, int, int, int]
 FaceLandmark203 = Tensor
 VisionTensor = Tensor
 LossTensor = Tensor
-GeneratorLossSet = Dict[str, LossTensor]
-DiscriminatorLossSet = Dict[str, LossTensor]
+GeneratorLossSet = Dict[str, Tensor]
+DiscriminatorLossSet = Dict[str, Tensor]
 VisionFrame = NDArray[Any]