diff --git a/embedding_converter/src/types.py b/embedding_converter/src/types.py
index 9221412..3a71965 100644
--- a/embedding_converter/src/types.py
+++ b/embedding_converter/src/types.py
@@ -3,11 +3,10 @@ from typing import Any, Tuple, TypeAlias
 from numpy.typing import NDArray
 from torch import Tensor
 
+Batch : TypeAlias = Tuple[Tensor, Tensor]
+
 Embedding : TypeAlias = NDArray[Any]
 EmbeddingDataset : TypeAlias = NDArray[Embedding]
 FaceLandmark5 : TypeAlias = NDArray[Any]
 
 VisionFrame : TypeAlias = NDArray[Any]
-VisionTensor : TypeAlias = Tensor
-
-Batch : TypeAlias = Tuple[VisionTensor, VisionTensor]
diff --git a/face_swapper/src/types.py b/face_swapper/src/types.py
index b5551d4..3a7bbed 100644
--- a/face_swapper/src/types.py
+++ b/face_swapper/src/types.py
@@ -5,6 +5,8 @@ from numpy.typing import NDArray
 from torch import Tensor
 from torch.nn import Module
 
+Batch : TypeAlias = Tuple[Tensor, Tensor, Tensor]
+
 ImagePathList : TypeAlias = List[str]
 ImagePathSet : TypeAlias = Dict[str, ImagePathList]
 
@@ -22,8 +24,6 @@ VisionFrame : TypeAlias = NDArray[Any]
 LossTensor : TypeAlias = Tensor
 VisionTensor : TypeAlias = Tensor
 
-Batch : TypeAlias = Tuple[VisionTensor, VisionTensor, Tensor]
-
 GeneratorLossSet : TypeAlias = Dict[str, Tensor]
 DiscriminatorLossSet : TypeAlias = Dict[str, Tensor]