diff --git a/embedding_converter/src/data_loader.py b/embedding_converter/src/data_loader.py
index 5585c5e..11f5a51 100644
--- a/embedding_converter/src/data_loader.py
+++ b/embedding_converter/src/data_loader.py
@@ -3,8 +3,8 @@ import random
 
 import cv2
 import torch
-import transforms from torchvision
 from torch.utils.data import Dataset
+from torchvision import transforms
 
 from .types import Batch, Paths
 
diff --git a/face_swapper/src/data_loader.py b/face_swapper/src/data_loader.py
index c562b8a..81f73a3 100644
--- a/face_swapper/src/data_loader.py
+++ b/face_swapper/src/data_loader.py
@@ -5,8 +5,8 @@ from typing import Tuple
 
 import cv2
 import torch
-import transforms from torchvision
 from torch.utils.data import TensorDataset
+from torchvision import transforms
 
 from .types import Batch, ImagePathList, ImagePathSet